GT_VTR3_1 / run /cloths_db.py
Ubuntu
fixed issuse with model with dress
6e6426e
cloths_map= {
'050105_1.jpeg': "Upper-body",
'02783_00.jpeg': "Upper-body",
'male_tshirt1.jpeg': "Upper-body",
'051515_1.jpeg': "Lower-body",
'07764_00.jpeg': "Upper-body",
'053786_1.jpeg': "Dress",
'09933_00.jpeg': "Upper-body",
'052234_1.jpeg': "Dress",
'049947_1.jpeg': "Upper-body",
'06123_00.jpeg': "Upper-body",
'051827_1.jpeg': "Lower-body",
'049805_1.jpeg': "Upper-body",
'049920_1.jpeg': "Upper-body",
'08348_00.jpeg': "Upper-body",
'050191_1.jpeg':"Upper-body",
'03244_00.jpeg': "Upper-body",
'11791_00.jpeg': "Upper-body",
'051946_1.jpeg': "Lower-body",
'00470_00.jpeg': "Upper-body",
'050181_1.jpeg': "Upper-body",
'03751_00.jpeg': "Upper-body",
'11028_00.jpeg': "Upper-body",
'07382_00.jpeg': "Upper-body",
'051998_1.jpeg': "Dress",
'051988_1.jpeg': "Lower-body",
'11351_00.jpeg': "Upper-body",
'07429_00.jpeg': "Upper-body",
'02305_00.jpeg': "Upper-body",
'03032_00.jpeg': "Upper-body",
'049949_1.jpeg': "Upper-body",
'049940_1.jpeg': "Upper-body",
'049965_1.jpeg': "Upper-body",
'053790_1.jpeg': "Dress",
'050002_1.jpeg': "Upper-body",
'10297_00.jpeg': "Upper-body",
'048769_1.jpeg': "Upper-body",
'00055_00.jpeg': "Upper-body",
'048554_1.jpeg': "Upper-body",
'051517_1.jpeg': "Lower-body",
'02015_00.jpeg': "Upper-body",
'053290_1.jpeg': "Dress",
'053742_1.jpeg': "Dress",
'04825_00.jpeg': "Upper-body",
'053319_1.jpeg': "Dress",
'053744_1.jpeg': "Dress",
'00151_00.jpeg': "Upper-body",
'06429_00.jpeg': "Upper-body",
'01260_00.jpeg': "Upper-body",
'12562_00.jpeg': "Upper-body",
'01430_00.jpeg': "Upper-body",
'051412_1.jpeg': "Lower-body",
'00126_00.jpeg': "Upper-body",
'051991_1.jpeg': "Lower-body",
'06802_00.jpeg': "Upper-body",
'051473_1.jpeg': "Lower-body"
}
modeL_db= {
'051962_0.jpg': "no-dress",
'052472_0.jpg': "dress",
'02783_00.jpg': "dress",
'09933_00.jpg': "dress",
'053228_0.jpg': "dress",
'051482_0.jpg': "no-dress",
'model_1.png': 0,
'051918_0.jpg': "no-dress",
'05997_00.jpg': 0,
'model_2.png': 0,
'08348_00.jpg': 0,
'07966_00.jpg': 0,
'11791_00.jpg': 0,
'model_9.png': "dress",
'03751_00.jpg': 0,
'11028_00.jpg': 0,
'14627_00.jpg': 0,
'11351_00.jpg': 0,
'053700_0.jpg': "dress",
'09597_00.jpg': 0,
'07429_00.jpg': 0,
'052964_0.jpg': "dress",
'model_3.png': 0,
'052767_0.jpg': "dress",
'049447_0.jpg': "no-dress",
'model_6.png': 0,
'model_7.png': 0,
'02849_00.jpg': 0,
'model_5.png': 0,
'model_4.png': 0,
'06429_00.jpg': 0,
'01260_00.jpg': 0,
'01008_00.jpg': "no-dress",
'049713_0.jpg': "no-dress",
'model_8.png': "no-dress",
'053514_0.jpg': "dress",
'01861_00.jpg': 0,
'01430_00.jpg': 0,
'06802_00.jpg': 0,
'049205_0.jpg': "no-dress"
}
# import os
# def list_files_in_folder(folder_path):
# """
# List all files in the given folder and create a dictionary with filenames as keys and 0 as values.
# Args:
# folder_path (str): Path to the folder.
# Returns:
# dict: Dictionary with filenames as keys and 0 as values.
# """
# files_dict = {}
# try:
# # List all files in the folder
# for file_name in os.listdir(folder_path):
# # Check if it's a file ("no-dress"t a directory)
# if os.path.isfile(os.path.join(folder_path, file_name)):
# files_dict[file_name] = 0
# except Exception as e:
# print(f"An error occurred: {e}")
# return files_dict
# # Example usage
# folder_path = 'run/examples/model'
# files_dict = list_files_in_folder(folder_path)
# print(files_dict)