Skip to content

Commit

Permalink
Update editorimg.py
Browse files Browse the repository at this point in the history
  • Loading branch information
uselessai authored May 5, 2023
1 parent 276bb84 commit 08ba59a
Showing 1 changed file with 31 additions and 11 deletions.
42 changes: 31 additions & 11 deletions editorimg.py
Original file line number Diff line number Diff line change
@@ -1,32 +1,52 @@
import glob
from PIL import Image
from os.path import basename
import os, sys
from pathlib import Path
import glob # Import glob module to find pathnames
from PIL import Image # Import Image module from PIL package to work with images
from os.path import basename # Import basename function from os.path module to get the base name of the file
import os # Import os module to work with the operating system
import sys # Import sys module to access some variables used or maintained by the interpreter and to interact with the interpreter
from pathlib import Path # Import Path module from pathlib package to work with directories and files paths

# CONVERTIMOS LAS IMAGESN A FORMATO JPG Y MODIFICAMOS EL TAMAÑO
# Define the original and the destination folder paths
originalesStylegan3 = (r'/home/laura/Escritorio/stylegan3/_noviembre2022')
destinoStylegan3 = (r'/home/laura/Escritorio/stylegan3/_noviembre22_resize/')

# Create a list of all the files and folders inside the original folder
contenido = os.listdir(originalesStylegan3)

# Create a list to store the names of all the images
imagenes = []

new_width = 64 #64
new_height = 128 #64
# Define the new height and width for the images
new_width = 64
new_height = 128

# Loop through all the folders in the original folder
for folders in contenido:
# Get the full path of the current folder
folder_path = os.path.join(originalesStylegan3, folders)

# Loop through all the files in the current folder
for fichero in os.listdir(folder_path):
# Check if the current file is a png file
if os.path.isfile(os.path.join(folder_path, fichero)) and fichero.endswith('.png'):
#imagenes.append(fichero)
# Open the current image file and resize it to the new dimensions
img = Image.open(os.path.join(folder_path, fichero))
img = img.resize((new_width, new_height), Image.ANTIALIAS)

# Get the current dimensions of the image
w, h = img.size

# Get the number at the start of the file name
fich = int(fichero.split("_")[0])

# Create a subfolder name using the number at the start of the file name
subfoldername = "2" + f"{fich:04d}"

# Create the subfolder if it does not already exist
if (not os.path.exists(destinoStylegan3 + subfoldername)):
os.makedirs(destinoStylegan3 + subfoldername)

# Save the image as a jpg file in the destination folder
img.crop((0, 0, w, h)).save(os.path.join(destinoStylegan3 + subfoldername, Path(fichero).stem + ".jpg"))

print(imagenes)

# Print the list of all the image names
print(imagenes)

0 comments on commit 08ba59a

Please sign in to comment.