深度学习模型试跑(三):detectron2

一.模型解读

今天是大名鼎鼎的detectron2,关于detectron2的解读数不胜数，在此我暂时就不推荐了！

官方代码

二.试跑效果

测试环境：

机器 : Dell T5820服务器
GPU: Nvidia P4000
cuda: 10.1
cudnn: 7.5
关键库: torch 1.3.0; torchvision 0.4.1; opencv-python 3.4.7
pycocotools 2.0.0(这个较低了，所以在训练后会报如下错,建议升级到最新的版本)

  File "/home/fisadm/.local/lib/python3.7/site-packages/pycocotools/coco.py", line 308, in loadRes
    if type(resFile) == str or type(resFile) == unicode:
Killedror: name 'unicode' is not defined

coco数据集训练的运行语句(一张GPU属实不给力,但是也很无奈!)

python tools/train_net.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml MODEL.WEIGHTS model_weights/R-50.pkl SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025

训练结果

试跑运行语句:

python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml --video-input Skate.mp4 --opts MODEL.WEIGHTS model_weights/mask_rcnn_X_101_32x8d_FPN_3x/model_final_2d9806.pkl

试跑视屏: