1. 首页
  2. 人工智能
  3. 机器学习
  4. 基于PyTorch、ONNX、TensorRT轻量化实现YOLOv4源码+权重文件+训练数据集

基于PyTorch、ONNX、TensorRT轻量化实现YOLOv4源码+权重文件+训练数据集

上传者: 2023-07-26 15:44:37上传 ZIP文件 719.67KB 热度 8次

基于PyTorch、ONNX、TensorRT轻量化实现YOLOv4源码+权重文件+训练数据集,详细目录结构如下:

├── README.md

├── dataset.py dataset

├── demo.py demo运行pytorch --> tool/darknet2pytorch

├── demo_darknet2onnx.py 转换为onnx格式的工具 --> tool/darknet2pytorch

├── demo_pytorch2onnx.py 转换为onnx格式的工具

├── models.py pytorch模型

├── train.py 训练models.py模型

├── cfg.py 训练所需的配置文件

├── cfg --> darknet2pytorch

├── data

用户评论