questions for inference.py

[meiguiz]

trafficstars

hello Dr.SeungBack, thanks for your nice project!

I want to make a category classification for your segmentation project,and I get the predicted result as shown above,but I do not know how to show the category name and labels in inference.py. Sorry to bother you.thank you very much!

[SeungBack]

Hi @meiguiz I think that adding the following lines at draw_prediction in utils/visualizer.py would show the category name.

labels = pred["labels"].detach().numpy()
instviz = imgviz.instances2rgb(image=rgb, masks=masks[cnd, :, :], labels=list(range(len(scores[cnd]))), 
                                    captions=[str(x) for x in labels[cnd]])

By the way, I am not a "Dr. "and a graduate student pursuing a Ph.D. degree....!