图片中类别的预测对应颜色如何同i

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Apache License 2.0

2.31k stars 259 forks source link

Closed todesti2 closed 8 months ago

todesti2 commented 8 months ago

好，在推理图片过程中，我发现图片中类别的预测对应颜色是随机的？每张图片同一类别的颜色不同，这似乎不太美观。但是我没有找到颜色的那个文件，通过搜COLOR我也不能确定。请问您可以指点一下设置颜色的文件在哪里吗？