hukaixuan19970627 / DOTA_devkit_YOLO

Trans DOTA OBB format(poly format) to YOLO format.
195 stars 39 forks source link

YOLO_Transform转换过程中解决出现图片占用问题(windows系统下) #21

Open z931761491 opened 1 year ago

z931761491 commented 1 year ago

最近在使用作者的YOLO_Transform.py转换DOTA数据格式时,如果是转换split后的图片的数据格式,因为split后有时会出现一个图片中gt为0的情况,这时需要删除该gt数目为0的图片和对应的label文件,此时出现了图片占用问题,需要修改一下代码。在dota2LongSideFormat这个方法中,将 img = Image.open(img_fullname)修改为 fp = open(img_fullname, 'rb') img = Image.open(fp) fp.close() 修改后运行正常,Linux下没试过,应该不会出现这种情况。

ulyduts commented 1 year ago

牛啊,解决了