-
PyTorch-YOLOv5-Mask_Detection下载
资源介绍
PyTorch-YOLOv5-Mask_Detection
●使用YOLOv5实现口罩附件检测
配置环境
执行如下代码:
$ pip3 install -r requirements.txt
下载数据集
图片: ://pan.baidu.com/s/1m-4Z80T4IBtJXrlJrRW9pg提取码:zpbw 标签: ://pan.baidu.com/s/1L-fYqoylbEKMyiKXCXz-7g提取码:szqg
口罩数据集预训练模型
链接: ://pan.baidu.com/s/1jQVOnqoy0HAswB67R8BQ0w提取代码:gypi将下载好的模型放在weights文件夹下。
数据处理
将下载好的图像和标签放在数据/ mydata文件夹下,train.txt和valid.txt我已经生成好放在mydata文件夹下了。
修改配置文件
找到data / coco128