-
PyTorch版本的YOLOv4,包含ONNX和TensorRT实现
资源介绍
pytorch-YOLOv4
YOLOv4的最小PyTorch实现。
Paper Yolo v4: :
源代码: :
更多详细信息: :
推理
火车
莫卡奇
├── README.md
├── dataset.py dataset
├── demo.py demo to run pytorch --> tool/darknet2pytorch
├── demo_darknet2onnx.py tool to convert into onnx --> tool/darknet2pytorch
├── demo_pytorch2onnx.py tool to convert into onnx
├── models.py model for pytorch
├── train.py