Skip to main content

One post tagged with "深度学习"

View All Tags

· One min read
老陈

必要资源


硬件配备

32G + 内存, 2060+ 英伟达显卡

操作步骤

1.准备好数据集, 使用labelme进行标注, 并转换为coco数据集

2.把yolov5 源码下载到本地电脑磁盘

3.查看训练yaml配置, 自己模仿写一个

4.查看数据集配置, 自己模仿写一个

5.开始训练

6.转换导出pt模型到onnx

7.onnx转换为ncnn

8.进行ncnn optimize优化

9.编辑ncnn(optimize)模型中的param文件, 如下图

使用Yyds.Auto加载模型

# 加载模型
model_yolo_reload("example.bin", "example.param")
# 以当前屏幕为目标, 进行前向推导计算
screen_yolo_locate()