推理图片的一种类别能否对应一种颜色

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我也不能确定。请问您可以指点一下设置颜色的文件在哪里吗？

KNzzzzzzzzzz commented 8 months ago

有推理图片的代码吗？怎么可视化

todesti2 commented 8 months ago

有推理图片的代码吗？怎么可视化

你可以看参考文档 python tools/infer.py那个

KNzzzzzzzzzz commented 8 months ago

我用的是pytorch的哪个版本，刚才看了下issue，解决了怎么推理 https://github.com/lyuwenyu/RT-DETR/issues/136#issuecomment-1876722312

todesti2 commented 8 months ago

我用的是pytorch的哪个版本，刚才看了下issue，解决了怎么推理 #136 (comment)

好的麻烦你删除一下你的问题好嘛哈哈