How to import a trained object detector to Simulink?

Hi. I have trained an object detector using trainYOLOv4ObjectDetector() function and now I want to add this detector to my Simulink model to detect the objects from the images I pass through in the Simulink model. Is there any way to import it? I tried using Deep learning object detector block but its not working. Could someone guide me through this?

Answer 1

[Ashu]

Hi Chandni,

You can use a MATLAB Function block in simulink to load the trained object detector .

To work with MATLAB Function block you can refer :

https://www.mathworks.com/help/simulink/ug/what-is-a-matlab-function-block.html

Inside the function you can have your custom code and make predictions like this

% Assuming trainedDetector.mat is the saved object detector

trainedDetector = load('trainedDetector.mat', 'trainedDetector')

I = imread('highway.png');
[bboxes, scores, labels] = detect(trainedDetector,I,Threshold=0.05);
detectedImg = insertObjectAnnotation(I,"Rectangle",bboxes,labels);

To read more about loading different types of variables, you can refer

https://www.mathworks.com/help/matlab/ref/load.html

To learn more about YoloV4ObjectDetector, you can refer

https://www.mathworks.com/help/vision/ref/trainyolov4objectdetector.html