目前代码里的目录检测用的yolov5的pt文件的标签好像固定的80个标签，如果我想自己指定标签应该怎么做？

ucla-mobility / OpenCDA

A generalized framework for prototyping full-stack cooperative driving automation applications under CARLA+SUMO.

Other

1k stars 198 forks source link

目前代码里的目录检测用的yolov5的pt文件的标签好像固定的80个标签，如果我想自己指定标签应该怎么做？ #169

Closed zandb2009 closed 2 years ago

zandb2009 commented 2 years ago

目前代码里的目录检测用的yolov5的pt文件的标签好像是固定的80个标签，我自己训练出来的模型不是按这80个标签来的，只有车辆，行人等标签，请问应该在哪里修改，才能让我的模型用例子的方法跑起来也不撞车？例子用的：python opencda.py -t single_town06_carla -v 0.9.13 --apply_ml

DerrickXuNu commented 2 years ago

在ml_manager.py里

zandb2009 commented 2 years ago

已经测试可以了