Al procesar imágenes de aprendizaje automático, debe colocar las imágenes en diferentes carpetas según la categoría, y también desea cambiar el nombre de las imágenes con incrementos de números debajo de las carpetas para facilitar el procesamiento posterior.
Primero, echemos un vistazo a la función de cambio de nombre de Python:
os.rename (src, dst)
src: el nombre del directorio que se va a modificar
dst: el nombre del directorio modificado
Si dst es un directorio existente, se lanzará OSError.
Nota: Tanto src como dst son la ruta completa + el nombre del archivo.
Primero, cambiemos el nombre de la carpeta
primero. Intente primero el método más simple:
import os
def myrename(path):
file_list=os.listdir(path)
i=0
for fi in file_list:
old_name=os.path.join(path,fi)
new_name=os.path.join(path,str(i))
os.rename(old_name,new_name)
i+=1
if __name__=="__main__":
path="D:/test/121"
myrename(path)
Podemos ver que las carpetas se han renombrado en orden, con un ligero cambio, parece no tan bajo:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in enumerate(file_list):
old_name=os.path.join(path,fi)
new_name=os.path.join(path,"N0."+str(i))
os.rename(old_name,new_name)
if __name__=="__main__":
path="D:/test/121"
myrename(path)
El efecto es el mismo, el rol de la función enumerar no se repetirá, también podemos usar python zip para modificar:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in zip(range(len(file_list)),file_list):
old_name=os.path.join(path,fi)
new_name=os.path.join(path,"The."+str(i))
os.rename(old_name,new_name)
if __name__=="__main__":
path="D:/test/121"
myrename(path)
El efecto es el mismo: a
continuación, cambiamos el nombre del archivo. Los tres métodos anteriores están disponibles y solo deben cambiarse ligeramente. Usaré la función enumerar como ejemplo para demostrar:
primero cree un documento de texto, copie y pegue un montón, como Para probar, agregue un try-except a la función para evitar errores de cambio de nombre de archivo. Aquí, el cambio de nombre debe ser un archivo del mismo formato; de lo contrario, se informará un error:
import os
def myrename(path):
file_list=os.listdir(path)
for i,fi in enumerate(file_list):
old_dir=os.path.join(path,fi)
filename="my"+str(i+1)+"."+str(fi.split(".")[-1])
new_dir=os.path.join(path,filename)
try:
os.rename(old_dir,new_dir)
except Exception as e:
print(e)
print("Failed!")
else:
print("SUcess!")
if __name__=="__main__":
path="D:/test/121"
myrename(path)
El código es muy simple, principiantes principiantes, aquí está el método más simple, ¡bienvenidos a intercambiar y discutir!