PyTorch深度学习计算机视觉框架
2024-09-02 09:44:16
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性能对比
物体检测
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 输出结果
最新文章
- laypage分页
- 驱动编程:NtReadVirtualMemory
- 【004: gcc 和 clang 的简单比较】
- 基于百度翻译API开发属于自己的翻译工具
- TCP - 流量控制 and 拥塞控制
- 保利威视Polyv点播集成
- JS中的forEach、$.each、map方法
- 1007: [HNOI2008]水平可见直线
- Docker网络管理-外部访问容器
- Matlab的parfor并行编程
- 剑指offer-二叉树的下一个节点
- AI LeNet
- xcode pod install 安装失败,提示缺少文件
- javascript DOM基础操作
- TensorFlow安装时错误CondaValueError: prefix already exists: G:\softs\Anaconda\envs\tensorflow
- [Web 前端] 使用yarn代替npm作为node.js的模块管理器
- codeforces 603 A
- MongoDB删除数据库
- Django和Ajax
- 一次spring aop 切面的问题
热门文章
- HDU 1210 Eddy's 洗牌问题(找规律,数学)
- Oracle递归 start with...connect by...prior
- 嵌入式:指针的指针、链表、UCOS 的 OSMemCreate 。
- 浅谈箭头函数和setTimeout中的this
- JavaScript 基础(六) 数组方法 闭包
- 监控tomcat工具使用
- ABAP术语-Implementation
- MySql指令的执行顺序
- Hadoop(23)-Yarn资源调度器
- python学习之路2(程序的控制结构)