PyTorch深度學習計算機視覺框架
Taylor Guo @ Shanghai - 2018.10.22 - 星期一
PyTorchCV" target="_blank" rel="nofollow,noindex">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效能對比
物體檢測
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