我正在使用下面链接中的克隆模型进行图像分割。
https://github.com/matterport/Mask_RCNN
我的系统详细信息是
- 处理器：Intel（R）Core（TM）i5-9600K CPU @ 3.7GHz
- RAM：8GB
- Windows：10
- Pro图形卡：GeForceGTX 1660
- CUDA：10.0
- cuDNN：7.6.1
- Python：3.7
- Tensorflow-gpu：1.14.0
型号详情：
Mask_RCNN
- 骨干网：ResNet-101
- 最大图像分辨率：480 x 320
- Steps_per_epoch：100
尽管我成功安装了tensorflow-gpu版本，但神经网络需要2天的训练时间。由于我是这个平台的新手，所以我不知道是否错过了任何事情。