文件名称:sharpmask:DeepMask和SharpMask的TensorFlow实现
文件大小:1.68MB
文件格式:ZIP
更新时间:2024-06-07 16:01:25
python machine-learning tensorflow ml image-recognition
锋利的面具 DeepMask 和SharpMask 在Python / TensorFlow上的实现。 介绍 该存储库包含DeepMask和SharpMask模型的实现。 DeepMask模型可预测与类无关的对象蒙版和对象得分,如果将对象居中并完全包含在图像中,则该值为正。 SharpMask是DeepMask体系结构的扩展,该体系结构使用自上而下的优化模块来计算更精确的对象蒙版建议。 该实现基于TensorFlow官方ResNet-v2 模型实现,并且需要预先训练的ResNet 。 ResNet模型的实现是从官方TensorFlow复制而来的。 请注意,ResNet模型检查点中的变量已显式放置在GPU设备上。 因此,它们将不会在CPU上还原。 快速开始 克隆此存储库git clone https://github.com/aby2s/sharpmask.git 从下载COCO
【文件预览】:
sharpmask-master
----requirements.txt(158B)
----run_model.py(4KB)
----sharpmask.py(19KB)
----LICENSE(11KB)
----resnet_model.py(22KB)
----README.md(4KB)
----prepare_data.py(6KB)
----example.png(1.65MB)
----im_classes.py(30KB)