![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
walletiger
这个作者很懒,什么都没留下…
展开
-
yolov5 pytorch-cpu 环境安装 及验证 (ubuntu20.04)
伸手党一枚,传奇算法撸起。。 一 系统环境: 主机:Intel(R) Core(TM) i7-8700 CPU @ 3.20GHz , 虚拟机环境 分配 6 cpu cores 内存: 16G 系统:Ubuntu 20.04.1 LTS (virtual box )虚拟机 二 本篇目标: 搭建 yolov5 训练环境 (cpu 版本) 后续计划: 训练 人手识别模型 部署到 jetson nano 环境 三 yolov5 需要的pytorch环境...原创 2020-12-06 14:05:58 · 3619 阅读 · 3 评论 -
用 yolov5s 训练口罩检测模型,并在 jetson nano 下部署测试
一 训练环境: ubuntu 20.04 + pytorch 1.7 + torchvison 0.8 gpu: gtx1080 8GB 二 下载 yolov5 模型及权重: https://github.com/ultralytics/yolov5 利用 weigths/download_weights.sh 来下载 yolov5s.pt , 必要的时候 科鞋上网 三 组织 训练数据 3.1 数据集下载 (感谢学长): https://pan.baidu.com/s/1...原创 2020-12-06 13:10:49 · 4685 阅读 · 12 评论