yolov5-4.0转caffe记录

最新推荐文章于 2025-03-07 14:59:59 发布

yayalejianyue

最新推荐文章于 2025-03-07 14:59:59 发布

阅读量4.3k

点赞数 7

文章标签： caffe linux 深度学习 pytorch 神经网络

本文链接：https://blog.csdn.net/yayalejianyue/article/details/114878503

版权

本文详细记录了如何在Ubuntu 16.04环境下，从安装基本库到将PyTorch的yolov5-4.0模型转换成Caffe模型的过程，包括环境配置、yolov5_caffe的编译与运行、onnx模型的转换等步骤，旨在帮助读者顺利完成模型迁移。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2.2.修改Makefile.config

2.3.修改Makefile

2.4安装yolov5_caffe环境命令

三、yolov5-4.0安装训练

四、onnx模型转换caffe模型

前言

记录一下安装yolov5_caffe环境以及将pytorch模型->onnx模型->caffe模型

首先感谢以下作者：

https://blog.csdn.net/Yong_Qi2015/article/details/114362223

https://blog.csdn.net/tangshopping/article/details/110470050

一、准备环境

1.1基本环境要求

nvidia/cuda:10.0

cudnn:7

ubuntu:16.04

1.2安装库文件，确保每一个都安装成功

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git
sudo apt-get install libprotobuf-dev libleveldb-dev libopencv-dev libsnappy-dev
sudo apt-get install libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install python-protobuf

1.3 安装Opencv

unzip OpenCV-3.4.5.zip
cd opencv

mkdir build && cd build
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
make -j${proc}
make install

查看是否opencv成功：