search

chaneyddtt / UDA-Animal-Pose

MIT License
102 stars 12 forks source link

Please include these lines at the end of get_cropped_TigDog.py file #16

Open jasorsi13 opened 1 year ago

jasorsi13 commented 1 year ago

Only after adding the lines the command "python get_cropped_TigDog.py" will run.

append file get_cropped_TigDog.py: ################################################

if __name__== "__main__":
    animals = ['horse', 'tiger']
    img_folder = './animal_data/'
    for animal in animals:
        img_list, anno_list = load_animal(data_dir=img_folder, animal=animal)
        img_idxs = dataset_filter(anno_list)
        print(len(img_idxs))

        if not os.path.exists(os.path.join(img_folder, 'real_animal_crop_v4',  'real_'+animal+'_crop')):
            os.makedirs(os.path.join(img_folder, 'real_animal_crop_v4', 'real_'+animal+'_crop'))

        get_cropped_dataset(img_folder, img_list, anno_list, img_idxs, animal)

###################################################

chaneyddtt commented 1 year ago

Hi @jasorsi13, sorry for the late reply. Yes, thanks for pointing out, pls add in the lines if you want to preprocess data by yourself.