京东金融大数据竞赛猪脸识别-从视频提取图像

Posted 2020-10-27

2017年11月的京东金融大数据竞赛参与人数最多的是猪脸识别的算法比赛，参加整个大数据比赛的有四千多人，而猪脸识别算法组就有一千多人。可见，搞图像识别的的人很多啊。想要提升自己价值的小伙伴们，向语音、文本、机器人等领域进发吧，有了机器学习的基础，入门这些领域应该都不是太难。
比赛给的数据是30头猪的视频，做识别的第一步还是从视频中去图像吧。本想用以前写过的视频取帧程序。看Matlab示例的时候发现用Matlab取帧更简便易行，那就用它吧。这样又省了很多时间。代码如下：

%exam1.m extract frame from video
clc;   
close all;   
imtool close all;  
clear; 
%30段视频在本代码同级目录video文件夹下，格式为mp4
fpath = [pwd,‘\video\*.mp4‘];
fname = dir(fpath);
fnum = numel(fname);
vidpath = [pwd,‘\video‘];
try  
    for i=1:fnum
        videoname = fullfile(vidpath,fname(i).name);
                %读取视频数据
        videoObject = VideoReader(videoname);
        % 获取视频中帧数.
        numberOfFrames = videoObject.NumberOfFrames; 
        vidHeight = videoObject.Height;
        vidWidth = videoObject.Width;
        numberOfFramesWritten = 0;

        figure;        
        writeToDisk = true;   
        %对视频文件全路径进行分离.
        [folder, baseFileName, extentions] = fileparts(videoname);  
        %获取当前路径       
        folder = pwd;
        %在当前路径下建立子文件夹image，并在image下为各视频建立以文件名命名的子文件夹，用以存放该视频提取的帧
        outputFolder = sprintf(‘%s/image/%s‘, folder, baseFileName);        
        if ~exist(outputFolder, ‘dir‘)
            mkdir(outputFolder); 
        end  
        %每分钟约为30帧图像，每隔15帧取一帧                
       for frame = 1 :15: numberOfFrames  
            %读取帧数据
            thisFrame = read(videoObject, frame);
            image(thisFrame);  
            caption = sprintf(‘Frame %4d of %d.‘, frame, numberOfFrames);  
            title(caption, ‘FontSize‘, fontSize);  
            drawnow; % Force it to refresh the window.  
            %将图像帧数据写入文件  
            if writeToDisk  
                % 创建图像名，格式为“视频名-帧序号”
                imgName = sprintf(‘%s-%04d.jpg‘,baseFileName, frame);  
                outputFullFileName = fullfile(outputFolder, imgName);      
                text(5, 15, imgName, ‘FontSize‘, 20);  
                % Extract the image with the text "burned into" it.  
                frameWithText = getframe(gca);  .  
                imwrite(frameWithText.cdata, outputFullFileName, ‘jpg‘);  
            end  

            % 输出提取信息.  
            if writeToDisk  
                progressIndication = sprintf(‘Wrote frame %4d of %d.‘, frame, numberOfFrames);  
            else  
                progressIndication = sprintf(‘Processed frame %4d of %d.‘, frame, numberOfFrames);  
            end  
            disp(progressIndication);      
            numberOfFramesWritten = numberOfFramesWritten + 1;  
       end  
    end
catch
        warning(‘Problem using function.  Assigning a value of 0.‘);
        a = 0;
end  

该程序运行完后，就得到了30头猪的图像信息。若需增加数据量，可将提取帧的间隔减小。