文件名称:pytorch-YOLOv4:YOLOv4的PyTorch,ONNX和TensorRT实现
文件大小:719KB
文件格式:ZIP
更新时间:2024-04-08 02:01:55
pytorch tensorrt onnx yolov3 yolov4
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
【文件预览】:
pytorch-YOLOv4-master
----.gitignore(131B)
----demo_trt.py(7KB)
----requirements.txt(158B)
----data()
--------giraffe.jpg(374KB)
--------prediction.jpg(77KB)
--------dog.jpg(160KB)
--------voc.names(135B)
--------coco.names(625B)
----DeepStream()
--------nvdsinfer_custom_impl_Yolo()
--------Readme.md(504B)
--------labels.txt(621B)
--------config_infer_primary_yoloV4.txt(3KB)
--------deepstream_app_config_yoloV4.txt(4KB)
----cfg.py(2KB)
----cfg()
--------yolov4-tiny.cfg(3KB)
--------yolov4-custom.cfg(12KB)
--------yolov3-tiny.cfg(2KB)
--------yolov3.cfg(8KB)
--------yolov4.cfg(12KB)
----models.py(17KB)
----demo_tensorflow.py(3KB)
----README.md(10KB)
----demo_darknet2onnx.py(2KB)
----Use_yolov4_to_train_your_own_data.md(3KB)
----License.txt(11KB)
----dataset.py(17KB)
----demo.py(4KB)
----tool()
--------darknet2onnx.py(3KB)
--------utils_iou_test.py(12KB)
--------__init__.py(0B)
--------utils_iou.py(7KB)
--------region_loss.py(9KB)
--------coco_annotation.py(2KB)
--------tv_reference()
--------utils.py(6KB)
--------onnx2tensorflow.py(1KB)
--------config.py(11KB)
--------camera.py(2KB)
--------yolo_layer.py(12KB)
--------darknet2pytorch.py(20KB)
--------torch_utils.py(3KB)
----demo_pytorch2onnx.py(3KB)
----evaluate_on_coco.py(12KB)
----train.py(27KB)