在Windows10环境下，使用TensorFlow-gpu 1.5.0和python 3.5.6训练Faster-RCNN时遇到两个错误：一是Incompatible shapes报错，二是image invalid, skipping警告。前者可能由于batch_size设置不当或数据输入问题，后者是由于proposal_target_layer.py中fg_inds和bg_inds数量不足导致。解决方案包括检查batch_size配置和调整roi_bg_threshold参数。"
103192038,7825642, JDBC数据库更新操作详解 ,"['后端开发', '数据库管理', 'JDBC', 'SQL']
1. 运行背景：Windows10，TensorFlow-gpu==1.5.0，python3.5.6。
2. Faster rcnn 源码来自https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5。
3. 训练前准备根据：
1）Windows 下 TensorFlow 和 Faster rcnn 详细安装步骤：
https://blog.youkuaiyun.com/tuoyakan9097/article/details/81782257#commentBox
2）Windows下使用Faster RCNN训练自己的模型：
https://blog.youkuaiyun.com/tuoyakan9097/article/details/82528245
（一）Incompatible shapes: [16,8] vs. [0,8]
我首先跑通了原始训练代码，第一次换成自己的图片也训练成功了。后来换成另一组数据（图片比较大），在之前换成自己的数据后的程序进行的修改，大致改了pascal_voc.py里的CLASS类别和config文件里把对应的max_size改大了，其余参数都没变。
在训练时报错：
Traceback (most recent call last):
File "C:\Users\think\Anaconda3\envs\t
最低0.47元/天 解锁文章
扫一扫
2602
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
