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

摘要： Taylor Guo @ Shanghai - 2018.10.22 - 星期一 PyTorch 資源連結 影象分類 VGG ResNet DenseNet Mobil...

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效能對比

物體檢測

• 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