43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
import os
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
# Definindo o diretório de origem das imagens
|
||
|
|
source_dir = 'dataset/images_o'
|
||
|
|
|
||
|
|
# Definindo o diretório de destino das imagens processadas
|
||
|
|
dest_dir = 'dataset/images'
|
||
|
|
|
||
|
|
# Verifica se o diretório de destino existe, senão cria
|
||
|
|
if not os.path.exists(dest_dir):
|
||
|
|
os.makedirs(dest_dir)
|
||
|
|
|
||
|
|
# Definindo o tamanho desejado para as imagens
|
||
|
|
desired_size = (512, 512)
|
||
|
|
|
||
|
|
# Inicializando o contador de imagens
|
||
|
|
image_counter = 1
|
||
|
|
|
||
|
|
# Listando todos os arquivos no diretório de origem
|
||
|
|
for filename in os.listdir(source_dir):
|
||
|
|
if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.tiff', '.bmp', '.gif')):
|
||
|
|
# Construindo o caminho completo do arquivo
|
||
|
|
file_path = os.path.join(source_dir, filename)
|
||
|
|
|
||
|
|
# Abrindo a imagem
|
||
|
|
with Image.open(file_path) as img:
|
||
|
|
# Redimensionando a imagem
|
||
|
|
img_resized = img.resize(desired_size)
|
||
|
|
|
||
|
|
# Construindo o novo nome do arquivo com o contador
|
||
|
|
# Isso garante que cada imagem tenha um nome único baseado em sua ordem de processamento
|
||
|
|
new_filename = f"{image_counter}.jpeg"
|
||
|
|
dest_path = os.path.join(dest_dir, new_filename)
|
||
|
|
|
||
|
|
# Salvando a imagem no formato JPEG
|
||
|
|
img_resized.save(dest_path, 'JPEG')
|
||
|
|
|
||
|
|
# Incrementando o contador após cada imagem processada
|
||
|
|
image_counter += 1
|
||
|
|
|
||
|
|
print("Processamento concluído.")
|