1运行检测结果
GPU_ID=0
CUDA_VISIBLE_DEVICES=${GPU_ID} ./tools/demo.py
2 ap map ;precision_recall:
./experiments/scripts/test_faster_rcnn.sh 0 pascal_voc vgg16
precision_recall参照博客修改
报错:
R = [obj for obj in recs[imagename] if obj[‘name’] == classname]
KeyError: ‘0000001’
这是在测试时出现错误,删掉VOCdevkit2007中annotations_cache的缓存
3 train
./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16
若训练过删除一下目录:
tf-faster-rcnn/tensorboard/vgg16(有很多个记录,可视化,可能看到不是正确结果)
tf-faster-rcnn/output/vgg16/voc_2007_trainval/default
tf-faster-rcnn/data/cache
tf_faster_rcnn/data/VOCdevkit2007/annotations_cache
修改configy.py中batchsize=256,train_faster_rcnn.sh中iter=110000,test_faster_rcnn.sh中iter=110000
4 可视化
迭代改为110000(原来70000)
batchsize改为256(原来128)。loss明显收敛
tensorboard --logdir=’/home/zbb/tf-faster-rcnn/tensorboard/vgg16_old/voc_2007_trainval’
default(测试集)
default_val(验证集)(中间跳了一下)