how can i modify this code for vehicle detection in images rather than in videos?

Question

saisps hmm 2017-7-30

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/350824-how-can-i-modify-this-code-for-vehicle-detection-in-images-rather-than-in-videos

回答： Image Analyst 2017-7-30

采纳的回答： Image Analyst

在 MATLAB Online 中打开

clc clear all clear

foregroundDetector = vision.ForegroundDetector('NumGaussians', 3, 'NumTrainingFrames', 50);

videoReader = vision.VideoFileReader('visiontraffic.avi'); for i = 1:150 frame = step(videoReader); % read the next video frame foreground = step(foregroundDetector, frame); end

figure; imshow(frame); title('Video Frame'); figure; imshow(foreground); title('Foreground'); se = strel('square', 3); filteredForeground = imopen(foreground, se); figure; imshow(filteredForeground); title('Clean Foreground');

blobAnalysis = vision.BlobAnalysis('BoundingBoxOutputPort', true, ... 'AreaOutputPort', false, 'CentroidOutputPort', false, ... 'MinimumBlobArea', 150); bbox = step(blobAnalysis, filteredForeground);

result = insertShape(frame, 'Rectangle', bbox, 'Color', 'green');

numCars = size(bbox, 1); result = insertText(result, [10 10], numCars, 'BoxOpacity', 1, ... 'FontSize', 14); figure; imshow(result); title('Detected Cars');

videoPlayer = vision.VideoPlayer('Name', 'Detected Cars'); videoPlayer.Position(3:4) = [650,400]; % window size: [width, height] se = strel('square', 3); % morphological filter for noise removal

while ~isDone(videoReader)

    frame = step(videoReader); % read the next video frame
    % Detect the foreground in the current video frame
    foreground = step(foregroundDetector, frame);
    % Use morphological opening to remove noise in the foreground
    filteredForeground = imopen(foreground, se);
    % Detect the connected components with the specified minimum area, and
    % compute their bounding boxes
    bbox = step(blobAnalysis, filteredForeground);
    % Draw bounding boxes around the detected cars
    result = insertShape(frame, 'Rectangle', bbox, 'Color', 'green');
    % Display the number of cars found in the video frame
    numCars = size(bbox, 1);
    result = insertText(result, [10 10], numCars, 'BoxOpacity', 1, ...
        'FontSize', 14);
    step(videoPlayer, result);  % display the results
end