Tianxiaomo/pytorch-YOLOv4

🚀 pytorch-YOLOv4: Phiên bản PyTorch của YOLOv4 với đầy đủ pipeline

Một implementation PyTorch nhẹ nhưng đầy đủ tính năng của YOLOv4, từ inference cho tới chuyển đổi ONNX và TensorRT.

pytorch-YOLOv4 là một dự án cài đặt YOLOv4 bằng PyTorch, hỗ trợ inference, huấn luyện (kể cả kỹ thuật mosaic), cùng chuyển đổi mô hình sang ONNX và TensorRT. Dự án tham chiếu trực tiếp từ darknet chính thức và đạt kết quả gần bằng với phiên bản gốc trên bộ đánh giá MS COCO.

✨ Điểm nổi bật

• Inference: Hỗ trợ chạy mô hình với trọng số darknet hoặc pytorch.
• Huấn luyện: Có hỗ trợ kỹ thuật mosaic để tăng hiệu quả học.
• Chuyển đổi darknet → pytorch: Công cụ darknet2pytorch giúp chuyển đổi trọng số từ darknet sang PyTorch.
• Chuyển đổi sang ONNX: Hỗ trợ xuất mô hình ra định dạng ONNX để triển khai trên các nền tảng khác nhau.
• Hỗ trợ TensorRT: Có thể chuyển đổi mô hình sang TensorRT để tối ưu tốc độ inference.
• Độ chính xác gần bằng darknet: Trên bộ đánh giá val2017, mô hình Pytorch đạt AP 0.466 so với 0.471 của darknet.

💡 Vì sao đáng chú ý

Khác với các implementation khác chỉ tập trung vào inference, dự án này cung cấp đầy đủ pipeline từ huấn luyện đến triển khai sản phẩm thực tế. Đặc biệt, hỗ trợ chuyển đổi liên qua ba nền tảng lớn: PyTorch, ONNX và TensorRT, giúp người dùng dễ dàng lựa chọn môi trường triển khai phù hợp. Kết quả đánh giá gần bằng với darknet chính thức cho thấy độ tin cậy cao của implementation này.

---

#YOLOv4 #PyTorch #ONNX #TensorRT #ObjectDetection #TiniX