标签:样本 裁剪 imgfile label strip 规则 path save 256
该代码是在处理DeepGlobe时所写,可灵活应用
deepglobe 数据训练图片和标签的大小都是1024×1024,而我们要处理成256×256;便可以使用下列代码。
def clip():
imgpath = "Original training image path"
labelpath = "Original label path"
imgfile = os.listdir(imgpath)
labelfile = os.listdir(labelpath)
conut = len(imgfile)
strip = np.int32(1024/256) #256 随自己变动
for i in range(conut):
imgp = imgpath + "/" + imgfile[i]
labelp = labelpath + "/" + labelfile[i]
imgname = imgfile[i][0:-4]
labelname = labelfile[i][0:-4]
img = Image.open(imgp)
label = Image.open(labelp)
saveimg = "save image path" + imgname
savelabel = "save label path" + labelname
c = 0
for j in range(strip):
for n in range(strip):
box = (256*j, 256*n, 256*(j+1), 256*(n+1)) #裁剪关键
img2 = img.crop(box)
label2 = label.crop(box)
si = saveimg + " %d" % (c+1) + ".png"
sl = savelabel + " %d " % (c+1) + ".png"
img2.save(si)
label2.save(sl)
c = c + 1
标签:样本,裁剪,imgfile,label,strip,规则,path,save,256 来源: https://blog.csdn.net/chen18876633032/article/details/117812493
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。