qinggangwu / yolov7-pose_Npoint_Ncla

基于yoloV7-pose添加任意个关键点和检测目标多分类
Other
112 stars 17 forks source link

我想咨询一下,如何训练多个类 #5

Open xs818818 opened 1 year ago

xs818818 commented 1 year ago

比如除了车牌,多识别一个车辆,围栏

qinggangwu commented 1 year ago

比如除了车牌,多识别一个车辆,围栏

多类别参考yoloV7系列的多类设置,nc设置为类别数量即可,如果该类别后面没有关键点,所有点的值都设置为-1即可。