代码拉取完成，页面将自动刷新
The YOLOv5 architecture is designed for efficient and accurate object detection tasks in real-time scenarios. It employs a single convolutional neural network to simultaneously predict bounding boxes and class probabilities for multiple objects within an image. The YOLOV5s is a tiny model.
# Install libGL
## CentOS
yum install -y mesa-libGL
## Ubuntu
apt install -y libgl1-mesa-dev
pip3 install tqdm
pip3 install onnx
pip3 install onnxsim
pip3 install ultralytics
pip3 install pycocotools
Pretrained model: https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s.pt
Dataset: http://images.cocodataset.org/zips/val2017.zip to download the validation dataset.
mkdir checkpoints
git clone https://github.com/ultralytics/yolov5
# 切换到需要的版本分支
git checkout v6.1
# 有一些环境需要安装
wget https://ultralytics.com/assets/Arial.ttf
cp Arial.ttf /root/.config/Ultralytics/Arial.ttf
# 转换为onnx (具体实现可以参考 export.py 中的 export_onnx 函数)
python3 export.py --weights yolov5s.pt --include onnx --opset 11 --batch-size 32
mv yolov5s.onnx /Path/to/checkpoints
export PROJ_DIR=/Path/to/yolov5s/ixrt
export DATASETS_DIR=/Path/to/coco2017/
export CHECKPOINTS_DIR=./checkpoints
export COCO_GT=${DATASETS_DIR}/annotations/instances_val2017.json
export EVAL_DIR=${DATASETS_DIR}/val2017
export RUN_DIR=${PROJ_DIR}/
export CONFIG_DIR=config/YOLOV5S_CONFIG
# Accuracy
bash scripts/infer_yolov5s_fp16_accuracy.sh
# Performance
bash scripts/infer_yolov5s_fp16_performance.sh
# Accuracy
bash scripts/infer_yolov5s_int8_accuracy.sh
# Performance
bash scripts/infer_yolov5s_int8_performance.sh
| Model | BatchSize | Precision | FPS | MAP@0.5 | MAP@0.5:0.95 |
|---|---|---|---|---|---|
| YOLOv5s | 32 | FP16 | 1112.66 | 0.565 | 0.370 |
| YOLOv5s | 32 | INT8 | 2440.54 | 0.557 | 0.351 |
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手
