吕文玉
吕文玉
这个是用的是PIL库做的 参考 utils/visualizer.py
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.6/ppdet/engine/trainer.py#L1111-L1118 参考这里的写法
是不是和你的cuda版本不一致?
模型config里全部的`num_classes`改成你想要的
短期简单的方法是在源码中找到相应的类,一般类的参数会有注释的
你的commit id是多少 , 我试了下原始的代码没问题 https://github.com/lyuwenyu/RT-DETR/ @that-learner @cqjnovo @wmjlincy
在检测中backbone最后一层的参数本来就不加载的,,那是训练分类时候用的
用的是那个模型 是官方提供的嘛
可以debug进去看下 label_to_cat_id_map 这个字典符合预期嘛,和你训练时候的类别一致不 https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.5/ppdet/metrics/json_results.py#L30