mask-rcnn-keras 一张图有100多个实例怎么提高速度训练啊太慢了

一张图有100多个实例怎么提高速度训练啊太慢了

Open dlml opened this issue 4 years ago • 5 comments

Dec 10 '20 05:12 dlml

啊这……你可以参考一下coco训练的那个数据加载方式，会快很多应该

Dec 14 '20 08:12 bubbliiiing

时间慢大主要原因是加载时间慢，主要问题在draw_mask函数上，里面有个三层循环构建mask的n层，n对应实例个数，对于实例过多数据集的训练，这里可以改一下

Oct 04 '21 08:10 sky1300

这个以后要改成coco数据集的格式了…之前不会装cocoapi，很多东西要重置，这个也是其中之一

Oct 05 '21 16:10 bubbliiiing

麻烦请教下，这个改成coco数据集了吗

Nov 05 '21 03:11 jiangxinufo

没……没想到我鸽了这么久，太少用mask rcnn了

Nov 08 '21 16:11 bubbliiiing