By the end of this post, you will learn how to: Train a SOTA YOLOv5 model on your own data. Sparsify the model using SparseML quantization aware training, sparse transfer learning, and one-shot quantization. Export the sparsified model and run it using the DeepSparse engine at insane speeds. P/S: The end result - YOLOv5 on CPU at 180+ FPS using on
HI, @dnth Thanks for the excellent work. Can I convert the ONNX or PyTorch model to tflite model? I roughly tried but I got the error.
After successful training and conversion, I got .pt and ONNX file. I convert the model using following command:
python3 yolov5-train/export.py --weights best.pt --include tflite --imgsz 320 --data my_data.yaml
HI, @dnth Thanks for the excellent work. Can I convert the ONNX or PyTorch model to tflite model? I roughly tried but I got the error. After successful training and conversion, I got .pt and ONNX file. I convert the model using following command:
python3 yolov5-train/export.py --weights best.pt --include tflite --imgsz 320 --data my_data.yaml
Output:
Any suggestion and help