使用Opencv4和YOLOv4(XTDrone)训练模型遇到问题的记录(二)
Written By PiscesAlpaca(双鱼座羊驼)
[TOC]
一、Opencv4安装问题记录
1.在cmake时,需要手动标记生成pgk-config文件
命令如下:
cmake -D CMAKE_BUILD_TYPE=Release -D OPENCV_GENERATE_PKGCONFIG=YES -D CMAKE_INSTALL_PREFIX=/usr/local/opencv4 ..
其中,OPENCV_GENERATE_PKGCONFIG=YES即为需要生成opencv4.pc文件,这一文件在之后进行YOLO模型训练时很重要,否则可能会出现找不到Opecv2及其头文件的情况,例如:
Package opencv was not found in the pkg-config search path.
Perhaps you should add the directory containing `opencv.pc'
to the PKG_CONFIG_PATH environment variable
No package 'opencv' found
./src/image_opencv.cpp:16:10: fatal error: opencv2/core/version.hpp: No such file or directory
16 | #include <opencv2/core/version.hpp>
| ^~~~~~~~~~~~~~~~~~~~~~~~~~
compilation terminated.
make: *** [Makefile:174: obj/image_opencv.o] Error 1
2.安装完成后需要配置pgk-config环境
使用如下命令查找opencv4.pc文件的位置
sudo find / -iname opencv4.pc
输出结果一般会有多个,例如:
/home/pisces/opencv/build/unix-install/opencv4.pc
find: ‘/run/user/1000/doc’: 权限不够
find: ‘/run/user/1000/gvfs’: 权限不够
/usr/local/opencv4/lib/pkgconfig/opencv4.pc
选择usr/local
开头的即为所需文件路径
将/usr/local/opencv4/lib/pkgconfig/
路径加入PKG_CONFIG_PATH
,使用命令(该文件可能为空文件):
sudo gedit /etc/profile.d/pkgconfig.sh
在打开的文件中键入如下语句:
export PKG_CONFIG_PATH=/usr/local/opencv4/lib/pkgconfig:$PKG_CONFIG_PATH
保存退出后,使用如下命令激活:
source /etc/profile
接着,验证是否配置成功,键入命令:
pkg-config --libs opencv4
输出形式为如下所示即为成功(根据个人环境而定,可以不同):
-L/usr/local/opencv4/lib -lopencv_gapi -lopencv_stitching -lopencv_alphamat -lopencv_aruco -lopencv_barcode -lopencv_bgsegm -lopencv_bioinspired -lopencv_ccalib -lopencv_cudabgsegm -lopencv_cudafeatures2d -lopencv_cudaobjdetect -lopencv_cudastereo -lopencv_dnn_objdetect -lopencv_dnn_superres -lopencv_dpm -lopencv_face -lopencv_freetype -lopencv_fuzzy -lopencv_hdf -lopencv_hfs -lopencv_img_hash -lopencv_intensity_transform -lopencv_line_descriptor -lopencv_mcc -lopencv_quality -lopencv_rapid -lopencv_reg -lopencv_rgbd -lopencv_saliency -lopencv_sfm -lopencv_stereo -lopencv_structured_light -lopencv_phase_unwrapping -lopencv_superres -lopencv_cudacodec -lopencv_surface_matching -lopencv_tracking -lopencv_highgui -lopencv_datasets -lopencv_text -lopencv_plot -lopencv_videostab -lopencv_cudaoptflow -lopencv_optflow -lopencv_cudalegacy -lopencv_videoio -lopencv_cudawarping -lopencv_viz -lopencv_wechat_qrcode -lopencv_xfeatures2d -lopencv_shape -lopencv_ml -lopencv_ximgproc -lopencv_video -lopencv_xobjdetect -lopencv_objdetect -lopencv_calib3d -lopencv_imgcodecs -lopencv_features2d -lopencv_dnn -lopencv_flann -lopencv_xphoto -lopencv_photo -lopencv_cudaimgproc -lopencv_cudafilters -lopencv_imgproc -lopencv_cudaarithm -lopencv_core -lopencv_cudev
继续配置动态苦环境,使得程序运行时可以加载.so
动态库文件,使用命令:
sudo gedit /etc/ld.so.conf.d/opencv4.conf
在打开的文件中键入如下语句:
/usr/local/opencv4/lib
保存退出后,执行以下命令使刚才配置的路径生效:
sudo ldconfig
二、darknet的问题记录
1.Couldn't find activation function mish, going with ReLU
XTDrone提供的darknet仓库和https://github.com/pjreddie/d...仓库在进行模型训练时都会遇到如下问题Couldn't find activation function mish, going with ReLU:
layer filters size input output
0 Couldn't find activation function mish, going with ReLU
conv 32 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 32 0.299 BFLOPs
1 Couldn't find activation function mish, going with ReLU
conv 64 3 x 3 / 2 416 x 416 x 32 -> 208 x 208 x 64 1.595 BFLOPs
2 Couldn't find activation function mish, going with ReLU
conv 64 1 x 1 / 1 208 x 208 x 64 -> 208 x 208 x 64 0.354 BFLOPs
需要替换为https://github.com/AlexeyAB/d...仓库的darknet即可解决
2.mosaic=1-compile Darknet with Opencv for using mosaic=1
替换~/catkin_ws/src/darknet_ros/darknet/cfg/xtdrone/obj_yolov4.cfg文件中:
#cutmix=1
mosaic=1
为
#cutmix=1
mosaic=0
3.CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertio `0' failed.
替换~/catkin_ws/src/darknet_ros/darknet/cfg/xtdrone/obj_yolov4.cfg文件中:
subdivisions=1
为
subdivisions=64
subdivision会让每一个batch不是一下子都丢到网络里。而是分成subdivision对应数字的份数,一份一份的跑完后,在一起打包算作完成一次iteration。这样会降低对显存的占用情况。如果设置这个参数为1的话就是一次性把所有batch的图片都丢到网络里,如果为2的话就是一次丢一半。
4.XTDrone提供的darknet库存在一些预先make过的文件,需要在本机重新make一遍,否则有些文件在本机无法找到。darknet也无需cmake,直接make就行。
5.make编译问题
darknet的MakeFile文件需要结合自己的GPU修改ARCH
,如:
ARCH= -gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=[sm_50,compute_50] \
-gencode arch=compute_52,code=[sm_52,compute_52] \
-gencode arch=compute_61,code=[sm_61,compute_61] \
-gencode arch=compute_75,code=[sm_75,compute_75]
需要结合GPU的运行能力,添加75 80 85等参数,参数30已经过时需要删除,35也即将过时。
三、使用YOLOv4进行训练时遇到的问题记录
1.无法找到opencv2头文件
正如上一节所示,在执行./darknet detector train cfg/xtdrone/obj.data cfg/xtdrone/obj_yolov4.cfg backup/yolov4.conv.137 -map
命令时(具体见XTDrone教程:https://www.yuque.com/xtdrone...),会出现无法找到opencv2头文件的情况,使用如下方法即可解决:
手动创建软链接命令如下:
sudo ln -s /usr/local/opencv4/lib/pkgconfig/opencv4.pc /usr/local/lib/pkgconfig/opencv4.pc
解释:由于darknet在寻找.pc
文件时,是在/usr/local/lib/pkgconfig
里寻找的,因此在该目录创建指向opencv4.pc文件的位置即可
此时yolo的训练过程就可以正常运行了
2.关于YOLO不使用GPU的问题
如果经过上述修改还是使用CPU ,则可能是darknet的MakefIle
文件中如下字段没有修改,均改为1重新编译即可:
GPU=1
CUDNN=1
OPENCV=1
四、参考资料
[1] Ubuntu 18.04安装OpenCV4.0和环境配置(重要): https://blog.csdn.net/new_del...
[2] windows下训练yolo时出现CUDA Error: out of memory问题的解决: https://blog.csdn.net/qq_3348...
[3] Couldn't find activation function mish, going with ReLU: https://github.com/pjreddie/d...
[4] ubuntu搭建darknet框架并使用alexeyAB版本的yolo跑demo: https://blog.csdn.net/weixin_...
部分参考资料或有遗漏,再次也对提供的帮助表示感谢,如有必要可联系我增加。
转载请注明出处!
本篇发布在以下博客或网站:
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。