search

bubbliiiing / mask-rcnn-keras

这是一个mask-rcnn的库，可以用于训练自己的实例分割模型。
MIT License
313 stars 70 forks source link

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

Open dlml opened 3 years ago

bubbliiiing commented 3 years ago

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

sky1300 commented 3 years ago

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

bubbliiiing commented 3 years ago

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

jiangxinufo commented 2 years ago

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

bubbliiiing commented 2 years ago

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