随笔分类 - YOLO
摘要:一:将PT模型文件转onnx: 1、用到export.py 安装依赖包 pip3 install onnx pip3 install coremltools==4.0 2、export.py中的配置 文件路径 模型类型 改好后直接点击运行,生成模型:
阅读全文
摘要:一、先用labelImag标注好自己的数据集: cmd输入:labelImg打开软件, 标注完后文件夹里的东西: 一张图片对应一个txt文档 文档里的内容: 还有一个总的类别文档classes.txt 训练集标注好后,再用同样的方法标注一些测试集图片,用于测试 标注好后将数据集放入项目根目录,一般数
阅读全文
摘要:1、安装python和PyTorch PyTorch官网 https://pytorch.org/ 2、下载yoloV5,并下载依赖文件 https://github.com/ultralytics/yolov5 解压后在文件夹里找到文件requirements.txt,里面列出了yolo5所需要的
阅读全文