机器学习、深度学习
文章平均质量分 77
主要是在学习人工智能时的一些记录
A Tiger
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
关于如何将pytorch模型转换为onnx及onnx转换为rknn
这里自己写了一个小程序生成dataset.txt,将dataset也放进刚刚的项目中。这里的dataset.txt是用来验证的数据集,就是数据图片的路径全都存在这里。然后运行就能生成在best.pt的目录下生成best.onnx文件。运行完后就会生成一个.rknn的文件。s中即为转化后的yoloV5的.txt数据集。将之前转换和划分好的数据集放入,修改yolo。JPEGImages里面存放着照片数据文件。Annotations里面存放着。为数据集中的训练集的比重,设置为8。文件参数,修改train。原创 2024-03-07 13:09:36 · 565 阅读 · 1 评论 -
使用yolov5进行训练的总结
自己在使用yolov5中遇到的一些问题原创 2024-03-07 12:55:08 · 782 阅读 · 1 评论
分享