Spaces:
Running
Running
File size: 1,549 Bytes
911c613 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import numpy as np
from PIL import Image
import os
def paddingjpg(path):
xmax = 0
ymax = 0
file_list = []
for FileNameLength in range(0, 100):
for dirName, subdirList, fileList in os.walk(path):
for filename in fileList:
# check file extension
if ".jpg" in filename.lower() and len(filename) == FileNameLength:
file_list.append(filename)
break
file_list.sort()
print(file_list)
temp_list = []
for filename in file_list:
image = Image.open(path + filename)
padded_image = Image.new("RGB", [560, 560])
padded_image.paste(image, (0,0))
padded_image.save(path + 'padded/' + filename)
def paddingpng(path):
xmax = 0
ymax = 0
file_list = []
for FileNameLength in range(0, 100):
for dirName, subdirList, fileList in os.walk(path):
for filename in fileList:
# check file extension
if ".png" in filename.lower() and len(filename) == FileNameLength:
file_list.append(filename)
break
file_list.sort()
print(file_list)
temp_list = []
for filename in file_list:
image = Image.open(path + filename)
padded_image = Image.new("L", [560, 560])
padded_image.paste(image, (0,0))
padded_image.save(path + 'padded/' + filename)
paddingjpg('../data/train/images/')
paddingpng('../data/train/labels/')
paddingjpg('../data/test/images/')
paddingpng('../data/test/labels/') |