PYTORCH
niuyuanye
这个作者很懒,什么都没留下…
展开
-
windows10+Pytorch中用yolov4.5测试
所谓的yolov5暂称为yolov4.5版本(但作为工程项目还是不错的,毕竟速度、精度、大小、部署都很友好):下载地址:https://github.com/ultralytics/yolov5 环境 2080显卡 windows10+pytorch1.5)+python3.7+CUDA:10.2+cuDNN7.6640*640 yolov5x(187M)版本 大概50ms 所有模型链接:https://download.csdn.net/download/niuyuanye/12538536640*...原创 2020-06-21 16:58:05 · 625 阅读 · 1 评论 -
windows10+MMdetection2.0的编译测试及faster_rcnn训练自己的数据
环境:windows10+ cundn10.1+ cudnn7.6+ vs2019+pythorch1.4或windows10+ cundn10.2+ cudnn7.6+vs2019+pythorch1.5 MMdetection2.0已于5.6发布:https://github.com/open-mmlab/mmdetection;这样其性能应该高于detectron2的性能(有待后续测试) 1、编译问题记录 需下载coco的pythonAPI mmcv库下载并安装 若直接用pip install mm.原创 2020-06-16 20:48:06 · 1918 阅读 · 1 评论 -
windows10+Pytorch中用yolov3训练自己数据记录
yolov3版本:下载地址:https://github.com/ultralytics/yolov3 选择原因pytorch中速度快、可以工业部署、不停的更新维护。后续有yolov5(只在其Github看到,后续可以尝试) 环境:windows10 pytorch1.5(1.4) python3.7 CUDA:10.2 (10.1) cuDNN7.6 Visual Studio Enterpris...原创 2020-06-12 17:48:06 · 2136 阅读 · 8 评论