FasterRCNN训练自己的数据
前言 因为科研需要,折腾了一阵Faster RCNN. 最开始将大佬chenyuntc的simple-faster-rcnn-pytorch鼓捣了一下使之能够在Windows下面跑(有需要的朋友可以联系我),但是该代码的batch size只能设为1,跑一次实验的耗时过长。本渣尝试花了四天时间通读了一遍大佬的代码,奈何功力尚浅,还做不到大范围的修改,因此从github上找到了jwyang的faster-rcnn.pytorch 稍作修改,在Linux下成功实验了自己的数据集。 代码 我所使用的服务器操作系统是Linux,安装了Python3.6和Pytorch1.3,并且服务器无法连接网络,导
用户评论