【CV5】Tensorflow_Fasterrcnn训练自己数据集
发布日期:2021-04-30 21:04:47 浏览次数:116 分类:精选文章

本文共 2399 字,大约阅读时间需要 7 分钟。

???????????

1. ????

??????????????????????? conda???????????

conda create -n tf-faster-rcnn python=3.6
conda remove -n tf-faster-rcnn --all # ???????
source activate tf-faster-rcnn
conda env list
source deactivate

??????????????

data/
??? VOCdevkit2007/
? ??? VOCtrainval_06-Nov-2007/
? ??? VOCtest_06-Nov-2007/
? ??? VOCdevkit_08-Jun-2007/
??? models/
??? resnet101/
??? vgg16/

????????

  • data??????
  • lib??????????????
  • tools??????????
  • voc_data?VOC????????pan.baidu.com/s/1UPRbgR2rMZUDHHGI6roGuA ????jlj8
  • pretrained_models??????????????pan.baidu.com/s/14kIWwxdOqCCioiyBDNtnlg ????gmta
  • setup.py??? GPU ? CPU ????

2. ?????????

1. ?? Cython

?????????????

make clean
make
pip install easydict
pip install Cython

????? python3 tool/demo.py ?????? ImportError: No module named gpu_nms??? Makefile ?? python ??? python3?

2. ?? COCO API

?????????????

cd datagit clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make

3. ?????

???????? VOC ????

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

???? VOCdevkit ?????? tf-faster-rcnn/data/VOCdevkit2007?

4. ????

????????? (voc_2007_trainval+voc_2012_trainval) ??? tf-faster-rcnn/output ???????

mkdir -p output/res101/voc_2007_trainval+voc_2012_trainval
ln -s ../../../voc_2007_trainval+voc_2012_trainval ./default

5. ????

???????????????

GPU_ID=0
CUDA_VISIBLE_DEVICES=${GPU_ID}
./tools/demo.py

3. ?? VGG16 ???????

1. ????

?? VGG16 ??????

wget http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz

2. ????

?? train_faster_rcnn.sh ?????

./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16

????????????? 20 ?????????

3. ????

??????????? output/vgg16/voc_2007_trainval/default/???????? output/vgg16/voc_2007_test/default/?

4. Tensorboard ???

?? Tensorboard ???????????

tensorboard --logdir=tensorboard/vgg16/voc_2007_trainval/ --port=7001 &

5. ?????

??????????????????

  • ?? MSDD ?????????? annotations_cache?results ? VOC2007 ????

  • ? VOC2007 ?????? Annotations?JPEGImages ? ImageSets ????

  • ? ImageSets ?????? Main ??????????????

  • ?????????? data/MSDD/VOC2007 ???????

  • ?? msdd.py ? factory.py ???????????? train_faster_rcnn.sh ??? msdd ??????

    6. ????

    ?? tools/demo.py ?? CLASSES?NETS ? DATASETS ???????????????????????????

    ???????????????????????

    上一篇:springboot解决form表单重复提交方案
    下一篇:指令整理

    发表评论

    最新留言

    路过,博主的博客真漂亮。。
    [***.116.15.85]2026年06月09日 00时50分04秒