C3D

来源：互联网发布：character mysql 编辑：程序博客网时间：2024/05/21 17:05
#include<iostream>  #include<stdio.h>  #include<vector>  #include<fstream>  #include<opencv2/opencv.hpp>  #define video_index_max 101  /************************************************************************函数功能：从已经获取ucf101数据集所有类名称的txt文件中读入视频类别名称，共101类输入参数：filename是.txt文件路径，videoname是保存101类的名称输出参数：************************************************************************/void ReadVideoClass(const std::string& filename, std::vector<std::string>&videoname){    std::fstream file0;    file0.open(filename);    if (!file0.is_open())    {        std::cerr << "fail to open the all name file" << std::endl;        exit(0);    }    char temp0[1000];    while (!file0.eof())    {        file0.getline(temp0, 1000);        std::string line(temp0);        videoname.push_back(line);    }    if (videoname.size() != video_index_max) //如果读入的所有名称数量小于101类，则输入文件有误      {        std::cerr << " video class file have error size" << std::endl;        exit(0);    }    file0.close();}/************************************************************************函数功能：获取所有的从C3D网络输出并保存的.prob文件的路径输入参数：filename是.txt文件路径，videoname是保存101类的名称输出参数：************************************************************************/void ReadAllTestResultPath(const std::string& filename, std::vector<std::string>&ResultPath){    std::fstream file1;    file1.open(filename);//读取视频文件路径      if (!file1.is_open())    {        std::cerr << "fail to open the test result file" << std::endl;        exit(0);    }    char temp[1000];    while (!file1.eof())    {        file1.getline(temp, 1000);        std::string line(temp);        //去除可能存在的空格，并添加后缀.prob          std::string PostfixLine;//测试结果文件保存路径          std::string::size_type npos0 = line.find_last_of(" ");        if (npos0 == line.size() - 1) {            PostfixLine = line.substr(0, npos0);            PostfixLine += ".prob";        }        else            PostfixLine = line + ".prob";        std::cout << PostfixLine << std::endl;        ResultPath.push_back(PostfixLine);    }    file1.close();}/************************************************************************函数功能：输入参数：输出参数：************************************************************************/void ReadLine(const std::string & path, std::string &Videopath, int &StartIndex, std::string& ground_truth){    std::string line = path;    StartIndex = atoi(line.substr(line.find_last_of("/") + 1, line.size() - 5).c_str());    ground_truth = line.substr(line.find("output/") + strlen("output/"), line.find("/v_") - strlen("output/"));    line.erase(0, 6); line.erase(line.find_last_of("/"), line.size());    line.insert(0, "F:/ucf101"); line += ".avi";    Videopath = line;}/************************************************************************函数功能：从.prob文件中读入C3D网络的输出结果输入参数：.prob文件路径输出参数：101类的预测输出结果 blob************************************************************************/bool read_blob(const std::string &probfile, std::vector<float>&blob){    /*读取模型提取的特征文件中的参数*/    std::ifstream fin(probfile, std::ios::binary);    if (!fin.is_open()) { std::cerr << probfile << "file not exist " << std::endl; return 0; }    int num = 0, channel = 0, length = 0, height = 0, width = 0;    fin.read((char*)&num, sizeof(int));    fin.read((char*)&channel, sizeof(int)); if (channel != video_index_max) return 0;    fin.read((char*)&length, sizeof(int)); fin.read((char*)&height, sizeof(int));    fin.read((char*)&width, sizeof(int));    float ptr = 0;    for (int i = 0; i < channel; i++)    {        fin.read((char*)&ptr, sizeof(float));        blob.push_back(ptr);    }    return 1;}/************************************************************************函数功能：从数组vec中找出最大值，返回最大值和最大值对应的序号输入参数：数组vec，数组大小num，引用返回的数组最大值rate输出参数：最大值对应的序号************************************************************************/int find_sec_max(std::vector<float>vec, int num, float &rate){    int index = -1;    if (vec.size() != num)        return -1;    float maxc = 0;    for (int i = 0; i < vec.size(); i++)    {        if (vec[i]>maxc)        {            maxc = vec[i]; index = i;        }    }    rate = maxc;    return index;}int main(){    using namespace std;    using namespace cv;    std::vector<std::string>videoname;    ReadVideoClass("output\\list.txt", videoname);    std::vector<std::string>ResultPath;    ReadAllTestResultPath("output\\last-test-output.lst", ResultPath);    VideoWriter writer("VideoTest.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(320, 240));    vector<string>TestResultAll;    vector<float>TestResultAllRate;    vector<string>GroundTrueAll;    for (size_t i = 0; i < ResultPath.size(); i++)    {        std::string Videopath; int StartIndex; std::string ground_truth;        ReadLine(ResultPath[i], Videopath, StartIndex, ground_truth);        std::vector<float>blob;        int flag = read_blob(ResultPath[i], blob);        if (!flag) { cerr << "blob read failed" << endl; exit(0); }        float maxrate = 0;        int TestResult = find_sec_max(blob, video_index_max, maxrate);        VideoCapture capture;        capture.open(Videopath);        int count = StartIndex;        int countcopy = count;        Mat frame;        while (countcopy--)//每次写视频时，都跳过前count帧开始写          {            capture >> frame;        }        while (count < StartIndex + 16)//每个视频段的大小都是16帧，          {            capture >> frame;//先读入至frame中              writer << frame;//将frame保存到writer中              TestResultAll.push_back(videoname[TestResult]);//每一帧都赋予一个检测结果（string类型），并将检测结果保存在TestResultAll中              TestResultAllRate.push_back(maxrate);//每一帧每一帧都赋予一个检测概率（float类型），并将检测概率值保存在TestResultAllRate中              GroundTrueAll.push_back(ground_truth);//每一帧都有一个ground truth               count++;        }    }    //writer.release();      VideoCapture writercapture; writercapture.open("VideoTest.avi");    VideoWriter VideoTest("VideoTestAll.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(320, 240));    int count = 0;    while (1)    {        Mat frame;        writercapture >> frame;        if (!frame.data)            break;        string result = TestResultAll[count];        float ratio = TestResultAllRate[count];        char stemp[200];        sprintf(stemp, "( %.3f )", ratio);        string truth = GroundTrueAll[count];        cv::putText(frame, "test result: " + result + stemp, cv::Point(5, 25), CV_FONT_HERSHEY_SIMPLEX, 0.5, cv::Scalar(0, 255, 255));        cv::putText(frame, "ground truth: " + truth, cv::Point(5, 48), CV_FONT_HERSHEY_SIMPLEX, 0.5, cv::Scalar(0, 0, 255));        imshow("frame", frame);        waitKey(30);        VideoTest << frame;        count++;    }    VideoTest.release();    return 0;}
阅读全文
0 0