diff --git a/utils/ImageUtils.py b/utils/ImageUtils.py index 03a5d34..2b359b1 100644 --- a/utils/ImageUtils.py +++ b/utils/ImageUtils.py @@ -62,28 +62,25 @@ class imageUtils: def splitimage(cls,src,rownum,colnum,dstpath): img=Image.open(src) w,h=img.size - if rownum<= h and colnum<=w: - s=os.path.split(src) - if dstpath=='': - dstpath = s[0] - if not os.path.exists(dstpath): - os.makedirs(dstpath) - fn=s[1].split('.') - basename=fn[0] - ext=fn[-1] - num=0 - rowheight=h//rownum - colwidth=w//colnum - for r in range(rownum): - for c in range(colnum): - box=(c*colwidth,r*rowheight,(c+1)*colwidth,(r+1)*rowheight) - count_image = "{:0>3d}".format(num) - file_path = os.path.join(dstpath,str(count_image)+'.'+ext) - print("file_path=",file_path) - img.crop(box).save(file_path) - num=num+1 - else: - print('不数!') + s=os.path.split(src) + if dstpath=='': + dstpath = s[0] + if not os.path.exists(dstpath): + os.makedirs(dstpath) + fn=s[1].split('.') + basename=fn[0] + ext=fn[-1] + num=0 + rowheight=h//rownum + colwidth=w//colnum + for r in range(rownum): + for c in range(colnum): + box=(c*colwidth,r*rowheight,(c+1)*colwidth,(r+1)*rowheight) + count_image = "{:0>3d}".format(num) + file_path = os.path.join(dstpath,str(count_image)+'.'+ext) + print("file_path=",file_path) + img.crop(box).save(file_path) + num=num+1 img.close return src