模型查看
模型预测
模型导出
模型训练完成后,找到训练文件生成文件夹,里面包含wights、过程图、曲线图。
模型预测
1、在以下文件夹中放入需要预测的图;
2、找到detect文件下的predict.py文件,修改以下内容。
3、右键点击运行,开始预测。预测结果如下。
模型导出
1、在engine文件夹下找到exporter.py文件,修改以下路径,并将default中的format格式改为onnx(第一次导出的话,可能需要自动下载onnx库)。
2、导出后,在文件夹中如下所示。