杜艳
PyTorch深度学习计算机视觉框架
来源:史东坡     发布时间: 2019-05-05      浏览次数:318

字号:

Taylor Guo @ Shanghai - 2018.10.22 - 星期一

 

PyTorch 资源链接

图像分类VGGResNetDenseNetMobileNetV2ResNeXtSqueezeNetShuffleNetShuffleNet V2位姿估计CPM: Convolutional Pose MachinesOpenPose: Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields物体检测SSD Faster R-CNNYoloV3FPN语义分割DeepLabV3PSPNetDenseASPP实例分割Mask R-CNN

 

PyTorchCV性能对比

物体检测

SSD:单窗多框检测器【SSD300, SSD512】

ModelTraining dataTesting datamAPFPS
SSD-300 OriginVOC07+12 trainvalVOC07 test0.772-
SSD-300 OursVOC07+12 trainvalVOC07 test0.786-
Faster R-CNN:区域建议网路 【Faster R-CNN】
ModelTraining dataTesting datamAPFPS
Faster R-CNN OriginVOC07 trainvalVOC07 test0.699-
Faster R-CNN OursVOC07 trainvalVOC07 test0.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 输出结果

 

  • 相关内容: