search

SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.41k stars 128 forks source link

Image contains class names #61

Closed oguzhannysr closed 1 year ago

oguzhannysr commented 1 year ago

hello, when I examined my output results, I saw that the class names are also in the image. How can I remove them from the image?

praeclarumjj3 commented 1 year ago

Hi @oguzhannysr, thanks for your interest. You can comment out all calls of self.draw_test(..) in demo/visualizer.py to remove the text from the output.

https://github.com/SHI-Labs/OneFormer/blob/7145cdaeea50968239055bcb0618d32bd306590f/demo/visualizer.py#L828-L834