PyTorch深度学习计算机视觉框架

当前位置:首页高博亚洲百家乐 >

高博亚洲百家乐

PyTorch深度学习计算机视觉框架

时间:2019-12-01本站浏览次数:150

       

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 输出结果

 




公司地址:(章贡区)赣州贸易广场(南康区)东山工业四路1号
联系人:魏香赞 13623816725
田继前 15336887075
电话:13937760969 传真:nlvukb@162.com
邮箱:j6txlc@hotmail.com

粤公网安备 44030702001579号

高博亚洲打不开@