PyTorch深度學習計算機視覺框架

Taylor Guo @ Shanghai - 2018.10.22 - 星期一

 

PyTorch 資源鏈接

• 圖像分類
  • 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
SSD-300 Origin	VOC07+12 trainval	VOC07 test	0.772	-
SSD-300 Ours	VOC07+12 trainval	VOC07 test	0.786	-

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

Model	Training data	Testing data	mAP	FPS
Faster R-CNN Origin	VOC07 trainval	VOC07 test	0.699	-
Faster R-CNN Ours	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 輸出結果