Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def get_pet_labels(pet_images):
- in_files = listdir("pet_images/")
- results_dic = dict()
- for idx in range[0, len(in_files), 1]:
- if in_files[idx][0] != ".":
- pet_label = ""
- pet_list += in_files[idx].lower().split("_")
- for word in pet_list:
- if word.isalpha():
- pet_label += word + " "
- pet_label = pet_label.strip()
- if in_files[idx] not in results_dic:
- results_dic[in_files[idx]] = [pet_label[idx]]
- else:
- print("** Warning: Duplicate files exist in directory:",
- in_files[idx])
- return results_dic
- for key in results_dic:
- print("Filename=", key, " Pet Label=", results_dic[key])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement