Weifeng-Chen / dl_scripts

MIT License
304 stars 122 forks source link

Update yolo2coco.py #6

Closed chenanga closed 1 year ago

chenanga commented 1 year ago

如果class name为traffic light这种形式的,中间有空格的话,原先的做法会把它分为两类,一个traffic类和一个light类。coco的80个类别中就有许多这种的。 现在这种做法把每一行的前后空格去除,至于中间是否有空格,都不会去理会,也就避免上述情况,只要classes.txt一行是一个类别,这样就没问题