tPretrainedDeeplebV3Plus.m

classdef(SharedTestFixtures = {DownloadDeeplabV3PlusFixture}) tPretrainedDeeplebV3Plus < matlab.unittest.TestCase
    % Test for tPretrainedDeeplebV3Plus
    
    % Copyright 2021 The MathWorks, Inc.
    
    % The shared test fixture downloads the model. Here we check the
    % inference on the pretrained model.
    properties        
        RepoRoot = getRepoRoot;
        ModelName = 'deepLabV3Plus-voc.mat';
    end
    
    methods(Test)
        function exerciseDetection(test)            
            model = load(fullfile(test.RepoRoot,'model',test.ModelName));
            image = imread('visionteam.jpg');
            sz = size(image);
            [~,k] = min(sz(1:2));            
            scale = 513/sz(k);
            img  = imresize(image, scale, "bilinear");
            imSize = size(img);
            imSize = imSize(:,1:2);
            actualLabel1Count = 252888;
            actualLabel2Count = 257034;
            
            result = semanticseg(img, model.net);            
            labelsCountTbl = countlabels(result(:));
            labelCount = labelsCountTbl.Count(find(labelsCountTbl.Count));
            
            % verifying size of output from semanticseg.
            test.verifyEqual(size(result),imSize);
            % verifying that all the pixels are labelled.
            test.verifyEqual(sum(labelCount),prod(imSize));
            % verifying the count of each labels on the result.
            test.verifyEqual(labelCount(1),actualLabel1Count);            
            test.verifyEqual(labelCount(2),actualLabel2Count);
        end       
    end
end