How to assign different color for different classes in object detection model?

8 次查看（过去 30 天）

Fahmi Akmal Dzulkifli 2023-4-9

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1944029-how-to-assign-different-color-for-different-classes-in-object-detection-model

评论： Fahmi Akmal Dzulkifli 2023-4-28

Greetings, I have two classes, which are labelled as normal and cancer cells. How to draw bounding boxes with different colors that referred to each class?

newp2 = imread([pathname filename]);
[bboxes,scores,labels] = detect(detector,newp2,'Threshold',0.7);
annotations = string(labels) + ": " + string(scores);
I = insertObjectAnnotation(newp2,'rectangle',bboxes,cellstr(annotations));
figure
imshow(I)

回答（1 个）

Sachin 2023-4-13

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1944029-how-to-assign-different-color-for-different-classes-in-object-detection-model#answer_1214913

在 MATLAB Online 中打开

I understand that you want to assign the different color for different classes in object detection model.

To draw bounding boxes with different colors for each class, you can modify the ‘Color’ parameter based on the class label,

Here is an example:

newp2 = imread([pathname filename]);
[bboxes,scores,labels] = detect(detector,newp2,'Threshold',0.7);
annotations = string(labels) + ": " + string(scores);
%loop through each detected object
for i=1:numel(bboxes)
  label = labels(i);
  bbox = bboxes(i,:);
  score = scores(i);
  
  if label=='inormal'
     color = 'b'
  elseif label == 'cancer'
  
    color = 'r'
  else 
    color = 'g'
insertObjectAnnotation(newp2,'rectangle',bboxes,cellstr(annotations),'Color',color);
 end

Please refer the following page for more information about ‘insertobjectannotation’

https://www.mathworks.com/help/vision/ref/insertobjectannotation.html#namevaluepairarguments

5 个评论
显示 3更早的评论隐藏 3更早的评论

Sachin 2023-4-22

Can you please tell what kind of error you are getting?

Fahmi Akmal Dzulkifli 2023-4-28

The error is the colour of the bounding boxes are same for all cells. Like in the above example, all the cells have green bounding boxes colour, includes the cancer cells. When I checked at the 'labels', it has two classes, but when to assign the colour, it only has one class only.

请先登录，再进行评论。

请先登录，再回答此问题。