本文共 2520 字，大约阅读时间需要 8 分钟。

???PaddlePaddle?????????

??????????????PaddlePaddle??????????????????????????????????????????

1. ????

1.1 ??PaddlePaddle-GPU

• ?????

  • 64?????
  • Python?>=2.7.15?3.x?
  • pip?>=9.0.1?
  • CUDA >= 9.0
  • cuDNN >= 7.6

• ?????

  • CUDA 9.0???pip install paddlepaddle-gpu==1.8.4.post97
  • CUDA 10.0???pip install paddlepaddle-gpu==1.8.4.post107
  • CPU???pip install paddlepaddle==1.8.4

• ?????

  import paddle.fluid as fluidfluid.install_check.run_check()

????????PaddlePaddle????????????1.8.4?

1.2 ??????

• COCO-API?
  • Windows?????????????C/C++???????????????Visual C++ 2015 x64 Native Build Tools Command?-?????pip install pycocotools

2. ????

2.1 ??PaddleDetection??

• ?????

  git clone https://github.com/PaddlePaddle/PaddleDetection.git

• ?????

  python -c "import paddle.fluid as fluid; fluid.install_check.run_check()"

?????????

2.2 ???????

• ?????

  python tools/infer.py -c configs/ppyolo/ppyolo.yml -o use_gpu=true weights=https://paddlemodels.bj.bcebos.com/object_detection/ppyolo.pdparams --infer_img=demo/000000014439.jpg

???????output????

3. ?????

3.1 ???????

• ???????????????

  VOCdevkit/??? Annotations/??? JPEGImages/??? ImageSets/?   ??? Main/?       ??? trainval.txt?       ??? test.txt

• ??trainval.txt?test.txt?

  import osfile_train = open('trainval.txt', 'w')file_test = open('test.txt', 'w')for xml_name in os.listdir(r'D:\computercode\python\PaddleDetection\dataset\insect\Annotations\train'):    file_train.write(xml_name[:-4] + '\n')for xml_name in os.listdir(r'D:\computercode\python\PaddleDetection\dataset\insect\Annotations\val'):    file_test.write(xml_name[:-4] + '\n')file_train.close()file_test.close()

4. ???????

4.1 ????

• ??Faster R-CNN?????

  configs/faster_rcnn_r50_fpn_1x.yml

• ?????
  • Backbone?ResNet50
  • FPN????
  • RPN head?BBoxHead??

4.2 ????

• ?????

  • max_iters: ????????batch_size??
  • milestones: ???????
  • base_lr: ?????

• ?????

  export CUDA_VISIBLE_DEVICES=0python -u tools/train.py -c configs/faster_rcnn_r50_fpn_1x.yml \    -o pretrain_weights=https://paddlemodels.bj.bcebos.com/object_detection/faster_rcnn_r50_fpn_1x.tar \    --eval

• ?????

  • ????????40??
  • mAP?0.50?11???71.60

5. ???????

5.1 ????

• ?????

  python -u tools/export_model.py -c configs/faster_rcnn_r50_fpn_1x.yml \    --output_dir=./inference_model

5.2 ????

• ?????

  python deploy/python/infer.py --model_dir=./inference_model/faster_rcnn_r50_fpn_1x \    --image_file=path/to/dataset/2572.jpeg --use_gpu=True

• ???????output????

6. ???????

6.1 ??????????

• ??????????????????????????

??????????

• ?????--eval?????
  • ??????Faster R-CNN?????????????????????????????1?

?????????????PaddlePaddle???????????????????

转载地址：http://vqch.baihongyu.com/