TX1上使用Faster-RCNN训练自己的训练集遇到的问题与解决

1.我使用的是嵌入式系统TX1，CPUGPU效率有限，只能使用ZF模型进行训练

2.数据集不能太小，第一次做数据集想只是试一试，只做了20个，结果费了半天劲，做不出效果，后来做了200+的数据集，才有了效果，做数据集不是很费事，建议做至少100+

3.我在训练时多次发生找不到"cv2"这个module的问题，简单的解决方法是在python安装目录下找到cv2.so文件，复制到出问题的文件夹即可解决问题，如下面报错的情况。

Traceback (most recent call last):
  File "demo.py", line 162, in <module>
    demo(net, im_name)
  File "demo.py", line 83, in demo
    im = cv2.imread(im_file)
AttributeError: 'module' object has no attribute 'imread'

4.由于显存有限，必须降低显存消耗，比如报错如下就是显存不足的原因

Process Process