前提
通常制作目标检测的训练集的时候需要对目标物进行标注,标注一般是用原图(像素很高,大小很大),人眼能看到更多的细节;但是标注完成后,进行训练的时候,受 GPU 的内存的限制,图片不能太大,要不内存溢出,会出现报错,一般可以通过调整 batch_size 来避免这个情况,但是batch_size 最小是1,如果还是出现 GPU 内存溢出,那么就需要修改图片的大小了,那么原来标注的 xml 文件的信息也要做出相应的调整
代码思路
1.读取xml文件(调用python的xml模块),保存原图的长宽
2.修改xml文件的<size></size>里面的<width>和