# Yolov5_for_PyTorch

**Repository Path**: wujianping996/Yolov5_for_PyTorch

## Basic Information

- **Project Name**: Yolov5_for_PyTorch
- **Description**: a copy of https://github.com/ultralytics/yolov5
- **Primary Language**: Python
- **License**: GPL-3.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 7
- **Created**: 2021-04-15
- **Last Updated**: 2021-04-15

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 1.版本说明
yolov5版本Tags=v2.0, python版本为3.7.5

# 2.配置数据集路径
修改data/coco.yaml文件中的train字段和val字段，分别指向当前环境中实际目录中的train2017.txt和val2017.txt，如：  
train: /data/coco/coco_yolo/train2017.txt  
val: /data/coco/coco_yolo/val2017.txt  

# 3.GPU,CPU依赖
按照requirements-GPU.txt安装python依赖包  

# 4.NPU依赖
除第3步依赖外，还需安装(NPU-driver.run, NPU-firmware.run, NPU-toolkit.run, torch-ascend.whl, apex.whl)

# 5.NPU 单机单卡训练指令  
bash train_npu_1p.sh  

# 6.NPU 单机八卡训练指令  
bash train_npu_8p.sh  

# 7.NPU evalution指令  
bash evaluation_npu_1p.sh  

# 8.GPU 单机单卡训练指令  
python train.py --data coco.yaml --cfg yolov5x.yaml --weights '' --batch-size 32 --device 0  

# 9.GPU 单机八卡训练指令  
python -m torch.distributed.launch --nproc_per_node 8 train.py --data coco.yaml --cfg yolov5x.yaml --weights '' --batch-size 256  

# 10.CPU指令  
python train.py --data coco.yaml --cfg yolov5x.yaml --weights '' --batch-size 32 --device cpu  

# 11.导出onnx指令
python export_onnx.py --weights ./xxx.pt --img-size 640 --batch-size 1