tensorflow - 通过 GTX 1080 微调 fast_resnet101_coco

Question

faster_rcnn_resnet101_coco是否可以通过GTX 1080with进行微调object detection api？或faster_rcnn_nasnet。

Answer 1

我不确定 1080 有多少 VRAM，但是您可以在 11GB RAM 的 1080ti 上训练更快的 rcnn resnet 101 模型。观察那里的 GPU 使用情况，它应该大致适合 8GB，批量大小为 1，所以我会说是的，您可以使用对象检测 api 微调 RCNN resnet101 对象检测器。