Taylor Guo @ Shanghai - 2018.10.22 - 星期一

• 影象分類
  • VGG
  • ResNet
  • DenseNet
  • MobileNetV2
  • ResNeXt
  • SqueezeNet
  • ShuffleNet
  • ShuffleNet V2
• 位姿估計
  • CPM: Convolutional Pose Machines
  • OpenPose: Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields
• 物體檢測
  • SSD 
  • Faster R-CNN
  • YoloV3
  • FPN
• 語義分割
  • DeepLabV3
  • PSPNet
  • DenseASPP
• 例項分割
  • Mask R-CNN

PyTorchCV效能對比

物體檢測

• SSD：單窗多框檢測器【SSD300, SSD512】

  • • Model	Training data	Testing data	mAP	FPS
      VOC07+12 trainval	VOC07 test	0.772	-
      VOC07+12 trainval	VOC07 test	0.786	-

• Faster R-CNN：區域建議網路 【Faster R-CNN】

• • Model	Training data	Testing data	mAP	FPS
    VOC07 trainval	VOC07 test	0.699	-
    VOC07 trainval	VOC07 test	0.706	-

• YoloV3

PyTorchCV 使用

以OpenPose為例：

• 訓練模型

python main.py  --hypes hypes/pose/coco/op_coco_pose.json \
                --base_lr 0.001 \
                --phase train \
                --gpu 0 1

• 微調模型

python main.py  --hypes hypes/pose/coco/op_coco_pose.json \
                --base_lr 0.001 \
                --phase train \
                --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
                --gpu 0 1

• 測試模型(test_img):

python main.py  --hypes hypes/pose/coco/op_coco_pose.json \
                --phase test \
                --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
                --test_img val/samples/ski.jpg \
                --gpu 0

• 測試模型(test_dir):

python main.py  --hypes hypes/pose/coco/op_coco_pose.json \
                --phase test \
                --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
                --test_dir val/samples \
                --gpu 0

PyTorchCV 結果例項

VGG19-OpenPose 輸出結果

VGG19-OpenPose 輸出結果