|
|
import PIL |
|
|
from PIL import Image |
|
|
import os, sys |
|
|
import time |
|
|
|
|
|
|
|
|
Width, Height = 480, 320 |
|
|
|
|
|
path1 = "DATASETs/wm5/" |
|
|
color_mode1 = "L" |
|
|
|
|
|
path2 = "DATASETs/cover5/" |
|
|
color_mode2 = "RGB" |
|
|
|
|
|
|
|
|
def resize(path, color_mode): |
|
|
dirs = os.listdir(path) |
|
|
print('before resize ', len(dirs)) |
|
|
for item in dirs: |
|
|
try: |
|
|
|
|
|
with Image.open(fr'{path}{item}') as im: |
|
|
resized = im.convert(f'{color_mode}').resize((Width,Height)) |
|
|
resized.save(fr'{path}{item}') |
|
|
time.sleep(0.0003) |
|
|
|
|
|
except PIL.UnidentifiedImageError: |
|
|
print(fr"Confirmed: This image {path}{item} cannot be opened!") |
|
|
|
|
|
except OSError: |
|
|
im = Image.open(fr'{path}{item}').convert(f'{color_mode}').resize((Width,Height)) |
|
|
im.save(fr'{path}{item}') |
|
|
print(fr"Chanched by hands for {path}{item}") |
|
|
dirs = os.listdir(path) |
|
|
print('after resize ', len(dirs)) |
|
|
|
|
|
|
|
|
resize(path1, color_mode1) |
|
|
resize(path2, color_mode2) |
|
|
|
|
|
|
|
|
def test_size(path): |
|
|
dirs = os.listdir(path) |
|
|
print('before test ', len(dirs)) |
|
|
for item in dirs: |
|
|
try: |
|
|
with Image.open(fr'{path}{item}') as im: |
|
|
width, height = im.size |
|
|
if (width == Width) and (height == Height): |
|
|
pass |
|
|
else: |
|
|
print(fr'for {item} not true size') |
|
|
time.sleep(0.0003) |
|
|
except PIL.UnidentifiedImageError: |
|
|
print(fr"Confirmed: This image {item} cannot be opened! We removed it") |
|
|
os.remove(f'{path}{item}') |
|
|
dirs = os.listdir(path) |
|
|
print('after test ', len(dirs)) |
|
|
|
|
|
|
|
|
test_size(path1) |
|
|
test_size(path2) |
|
|
|
|
|
|
|
|
def renameimg(path): |
|
|
os.getcwd() |
|
|
|
|
|
for i, filename in enumerate(os.listdir(path)): |
|
|
try: |
|
|
os.rename(path + "/" + filename, path + "/" + str(i) + ".jpeg") |
|
|
|
|
|
except FileExistsError: |
|
|
pass |
|
|
|
|
|
renameimg(path1) |
|
|
renameimg(path2) |
