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下面的路径。

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环境各种。

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

Error2：showimage参数不足

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

 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()

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