# Generate_Yolov5_Dataset

**Repository Path**: trisoill/Generate_Yolov5_Dataset

## Basic Information

- **Project Name**: Generate_Yolov5_Dataset
- **Description**: 快速生成yolov5数据集脚本
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2022-09-27
- **Last Updated**: 2022-11-10

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Gererate_Yolov5_Dataset 使用说明

## 环境

- 本项目测试运行于Macos 12.6，其余可运行shell脚本的系统环境暂未测试

## 使用方式

- 将需要生成的数据集的图片放于Images文件夹中
- 将使用lableImg生成的xml放于Annotation文件夹中
- 运行命令

```shell
sudo chmod +x run.sh
./run.sh
```

- 运行上述命令后，会生成dataset文件夹，将其拷贝至yolov5根目录下，cd至该目录后，运行：

```shell
python train.py --img 640 --batch 16 --epochs 150 --data dataset/data.yaml --cfg models/yolov5m.yaml --weights yolov5m6.pt --device cpu
```

- weights可换为其他预训练模型，注意的是需要同时修改--cfg参数

## 参考

- 本开源脚本参考：https://lncoder.blog.csdn.net/article/details/124860809