yzqxy / Yolov8_obb_Prune_Track

GNU General Public License v3.0
185 stars 15 forks source link

关于detect中的save_txt #12

Open pcycccccc opened 1 year ago

pcycccccc commented 1 year ago

我想看旋转目标模型输出的信息有哪些,设置了save_txt为true,想保存检测完的标签信息,但是运行的时候有错误 image 但是报错的上面显示了检测的信息,看起来应该是4个点的信息,这是边框的信息吗? 我的问题是, 1、该如何修改这个代码让其运行,看起来还是有点小bug的,不知道大佬那边是不是能成功运行 2、save_txt和模型检测头输出的信息是一样的吗?还是说save_txt只是保存检测的信息,模型检测头输出的是【类别、置信度、中心点坐标、长宽、旋转角度】

yzqxy commented 1 year ago

我想看旋转目标模型输出的信息有哪些,设置了save_txt为true,想保存检测完的标签信息,但是运行的时候有错误 image 但是报错的上面显示了检测的信息,看起来应该是4个点的信息,这是边框的信息吗? 我的问题是, 1、该如何修改这个代码让其运行,看起来还是有点小bug的,不知道大佬那边是不是能成功运行 2、save_txt和模型检测头输出的信息是一样的吗?还是说save_txt只是保存检测的信息,模型检测头输出的是【类别、置信度、中心点坐标、长宽、旋转角度】

这个也有技友给我提了,我之前没考虑到,就没写,后续给加上。你也可以自己写,根据网络输出,转成一开始读取gt的txt里面的格式。也就是x,y,w,h,θ,转成四个角点坐标。然后再缩放回原图大小生成xml保存,你就可以在rolabelimg里打开看了,理论上是这样的,写的话还是得花点功夫,所以你可以先尝试自己改,等我不忙的时候我再补上