yolov3+opencv+视频检测保存

目录

1.opencv的安装

1)去官网下载opencv

2)解压下载下来的zip包

 3)安装必须库和cmake-gui

4)cmake-gui编译

5)make和make install

6)路径设置

7)测试一下:

2.yolov3源码更改与排错

Error1:cv_window_normal未定义

Error2:showimage参数不足

3.测试视频与保存

1)官方命令:

2)python为接口


1.opencv的安装

之前感觉太麻烦放弃了,检测图片直接保存,没用到。现在想检测视频,是避不开了,弄了我一天才安装好,之前一直make失败,后来发现opencv4.1.1不可用,又换3.4.2心累,结果配置好了,环境变量不生效,重启了傻逼电脑进不去系统,一直紫屏,呜呜,最后大神都放弃,帮我拯救了,重启了一下好了,不过进入系统会闪屏,而且垃圾箱不可用,不过能进就好,呜呜。

建议使用cmake-gui编译,容易发现问题。

1)去官网下载opencv

在本教程中选用的时opencv3.4.2,其他版本的配置方法其实差不多。 

   下载链接http://opencv.org/releases.html,选择sources版本

2)解压下载下来的zip包

unzip opencv-3.4.2.zip

 3)安装必须库和cmake-gui

sudo apt-get install cmake-gui
sudo apt-get install build-essential libgtk2.0-dev libavcodec-dev libavformat-dev libjpeg.dev libtiff4.dev libswscale-dev libjasper-dev  

库不全的话,后期cmake有问题再补

4)cmake-gui编译

进入到解压后的文件包中,在解压的文件夹里新建一个文件夹build用来编译OpenCV

中间cmake编译过程,参考博客https://blog.csdn.net/jindunwan7388/article/details/80397700和

 https://blog.csdn.net/u011897411/article/details/89743448

中间解决ippicv下载问题,离线下载

# 打开终端,输入
 
    gedit ~/DownLoad/opencv_source/opencv/3rdparty/ippicv/ippicv.cmake 
 
# 将47行的
 
     "https://raw.githubusercontent.com/opencv/opencv_3rdparty/${IPPICV_COMMIT}ippicv/"
 
# 改为手动下载的文件的本地路径(也就是将网络下载的模式改为本地文件下载的模式):
 
    "file:///path" #(根据文件路径填写,注意是///)

5)make和make install

我就是make卡住了

make -j8

出现的错误:

/usr/bin/ld: warning: libiconv.so.2, needed by //home/smiles/anaconda2/lib/libgobject-2.0.so.0, not found (try using -rpath or -rpath-link)
/usr/bin/ld: warning: libpcre.so.1, needed by //home/smiles/anaconda2/lib/libgobject-2.0.so.0, not found (try using -rpath or -rpath-link)
//home/smiles/anaconda2/lib/libglib-2.0.so.0:对‘libiconv_open’未定义的引用
//home/smiles/anaconda2/lib/libglib-2.0.so.0:对‘libiconv_close’未定义的引用
//home/smiles/anaconda2/lib/libglib-2.0.so.0:对‘libiconv’未定义的引用 

解决:参照博客https://blog.csdn.net/tsq292978891/article/details/78854188

把cmake编译中的python3改为anaconda下面,并把python的包的路径,都改成了anaconda3下面的路径。

yolov3+opencv+视频检测保存_第1张图片

make install

6)路径设置

参考对应版本(3或4)博客https://blog.csdn.net/jindunwan7388/article/details/80397700和

 https://blog.csdn.net/u011897411/article/details/89743448

7)测试一下:

# 命令行输入以下命令
pkg-config --modversion opencv

2.yolov3源码更改与排错

Error1:cv_window_normal未定义

终于知道这些错误原因了,是根据博客https://blog.csdn.net/babyzbb636/article/details/100534359更改,版本不同,把对应部分改成源码对应或者对应下面排错。

差点因为这个原因就放弃了,各种百度找不到,找到的也不适用,还怀疑opencv环境各种。

yolov3+opencv+视频检测保存_第2张图片

最后想起来看源码,看哪里错误,发现,它好像不识别的是这几个定义; 查了一下这个函数的用法,想到了可能是版本差异:改成如下就好了,记住一共两处修改 

yolov3+opencv+视频检测保存_第3张图片

Error2:showimage参数不足

showimage源码参数是两个,这个也是opencv版本不同,改成三个,第三参数为0, 把cvSetwindowproperty注释了,有需要的网友可以根据版本更改

yolov3+opencv+视频检测保存_第4张图片

 Make -j8 success天呀

3.测试视频与保存

1)官方命令:

./darknet detector demo cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_10000.weights test_video/video/video001.avi

其中只能显示最后的结果无法保存,或者使用

./darknet detector demo cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_10000.weights test_video/video/video001.avi -prefix test_video/result/video001.mp4

最后是逐帧保存,也不太好,所以这里修改下源码:

1.在src/image.c里面修改show_image函数

int show_image(image p, const char *name, int ms)
{
#ifdef OPENCV
    // int c = show_image_cv(p, name, ms);
    // 增加保存结果视频
    int c = save_video(p, name, ms);
    return c;
#else
    fprintf(stderr, "Not compiled with OpenCV, saving to %s.png instead\n", name);
    save_image(p, name);
    return -1;
#endif
}

2.在src/image_opencv.cpp里面增加一个save_video函数

int save_video(image im, const char*name, int ms)
{
    // 静态数据成员,第一次调用初始化,之后调用不会在初始化
    static VideoWriter* video;
    Mat m = image_to_mat(im);
    // imshow(name, m);
    {
        // 空视频对象则初始化一个对象
        if(video == NULL){
            const char* output_name = "test_video/result/video001.mp4"; //修改输出路径
            // 新建视频保存对象
            video = new VideoWriter(output_name, VideoWriter::fourcc('M','J','P','G'), 50, Size(im.w,im.h));
            printf("\n DST output_video = %s  \n", output_name);
        }
        // 写入每一帧的处理结果
        video->write(m);
		printf("\n cvWriteFrame \n");
    }
    int c = waitKey(ms);
    if (c != -1) c = c%256;
    return c;
}

2)python为接口

参考文献:

https://blog.csdn.net/orDream/article/details/84311697

https://blog.csdn.net/dragongiri/article/details/90647089

https://blog.csdn.net/OliverkingLi/article/details/93487281

其中我把image.c和image.h的条件编译去掉了,否则老报libdarknet.so: undefined symbol: ndarray_to_image,可能我插入代码位置不对。终于实现了视频检测。

之前保存视频老打不开,在darknet.py的main中,贴出我修改后的代码(前面参照我贴出来的博客修改):

if __name__ == "__main__":
    net = load_net("/home/zbb/darknet/cfg/yolov3-voc.cfg".encode('utf-8'), "/home/zbb/darknet/backup/yolov3-voc_10000.weights".encode('utf-8'), 0)
    meta = load_meta("/home/zbb/darknet/cfg/voc.data".encode('utf-8'))
    in_path='/home/zbb/darknet/test_video/plane.MP4'
    #out_path='/home/zbb/darknet/test_video/result/test.avi'
    vid = cv2.VideoCapture(in_path)
    total_frames = vid.get(cv2.CAP_PROP_FRAME_COUNT)
    fps = vid.get(cv2.CAP_PROP_FPS)
    frame_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)), int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))
    fourcc = cv2.VideoWriter_fourcc(*"mp4v") #avi格式:*'XVID'  mp4格式: *"mp4v"
    out_path = in_path.split(".")[0] + "-result.mp4"
    videoWriter = cv2.VideoWriter(out_path, fourcc, fps, frame_size)
    #videoWriter = cv2.VideoWriter('/home/zbb/darknet/test_video/result/test.avi', fourcc, 25, (720,404)) #改成自己的framesize或者用上面的写入

    while True:
        return_value,arr=vid.read()
        if not return_value:
            break 
        im=nparray_to_image(arr)
        boxes= detect(net, meta, im)
       
        for i in range(len(boxes)):
            score=boxes[i][1]
            label=boxes[i][0]
            xmin=boxes[i][2][0]-boxes[i][2][2]/2
            ymin=boxes[i][2][1]-boxes[i][2][3]/2
            xmax=boxes[i][2][0]+boxes[i][2][2]/2
            ymax=boxes[i][2][1]+boxes[i][2][3]/2
            print(label)
            cv2.rectangle(arr,(int(xmin),int(ymin)),(int(xmax),int(ymax)),(0,255,0),2)
            cv2.putText(arr,str(label),(int(xmin),int(ymin)),fontFace=cv2.FONT_HERSHEY_SIMPLEX,fontScale=0.8,color=(0,255,255),thickness=2)    
        cv2.imshow("Canvas", arr)
        videoWriter.write(arr) 
        cv2.waitKey(1) 
    cv2.destroyAllWindows()

yolov3+opencv+视频检测保存_第5张图片

现在存在问题是标签不对,有b’,看了detect也没有问题,简单去掉'

你可能感兴趣的:(深度学习)