pytorch
关注
分享
扫一扫
ruixia.chen
从事深度学习工作
展开
-
WIN11安装detectron2踩坑
windows 环境安装detectron2详细步骤原创 2022-07-11 15:18:41 · 2044 阅读 · 1 评论 -
conda 配置虚拟环境
conda 环境配置原创 2022-06-14 16:25:33 · 1791 阅读 · 0 评论 -
yolov4效果展示
yolov4和yolov3性能比较：使用二维码数据测试：yolov3 单张图139ms，yolov4 单张图99ms，能快40ms。从bottle数据和二维码数据结果看，yolov4边框的回归也比yolov3回归的更好。yolov4可以完美的取代yolov3。使用yolov4代码训练的bottle和二维码的数据，大概训练一个小时，进行预测效果如下：...原创 2021-06-24 17:29:13 · 327 阅读 · 0 评论 -
提生pytorch dataloader数据加载速度
因为pytorch数据加载速度太慢，影响训练速度，实训快速加载数据方式，提前获取要加载的数据，整体速度能快1/6.操作步骤如下所示：1、激活自己的torch虚拟环境:source activate torch2、安装prefetch_generator包pip install prefetch_generator3、使用时候引入对应包：from prefetch_generatorimport BackgroundGenerator4、定义DataLoaderX，继承torc原创 2021-06-23 10:08:01 · 1387 阅读 · 1 评论