Go to file
fly6516 7e38b3acfd feat: 添加 YOLO 模型从 PyTorch 到 ONNX 的转换及验证功能
- 实现了将 YOLO模型从 PyTorch 格式转换为 ONNX 格式的功能
- 加载并验证了导出的 ONNX 模型
- 编写了使用验证集进行模型测试的代码,包括图像预处理和推理
- 添加了调试信息输出,便于分析模型性能和验证结果
2025-06-03 23:49:21 +08:00
datasets feat(datasets): 添加新的训练和验证标签文件 2025-06-03 23:13:32 +08:00
runs/detect feat: 添加 YOLO 模型从 PyTorch 到 ONNX 的转换及验证功能 2025-06-03 23:49:21 +08:00
data.yaml feat(datasets): 添加新的训练和验证标签文件 2025-06-03 23:13:32 +08:00
onnx2tflite.py feat(datasets): 添加新的训练和验证标签文件 2025-06-03 23:13:32 +08:00
pt2onnx.py feat: 添加 YOLO 模型从 PyTorch 到 ONNX 的转换及验证功能 2025-06-03 23:49:21 +08:00
test.py feat(datasets): 添加新的训练和验证标签文件 2025-06-03 23:13:32 +08:00
train.py feat(datasets): 添加新的训练和验证标签文件 2025-06-03 23:13:32 +08:00