Beispiel #1
0
 def get_features(self, label):
     test_dataset_folder_path = os.path.abspath(
         os.path.join(Path(os.getcwd()).parent, self.labelled_dataset_path))
     images_list = list(
         misc.get_images_in_directory(test_dataset_folder_path).keys())
     metadata = Metadata(images_list)
     if self.feature_name != 'SIFT':
         metadata.save_label_decomposed_features(label,
                                                 self.decomposed_feature)
         features = misc.load_from_pickle(
             self.reduced_pickle_file_folder,
             self.decomposed_feature + '_' + label)
     else:
         features = {}
         database_features = misc.load_from_pickle(
             self.main_pickle_file_folder, self.feature_name)
         label_images_list = metadata.get_specific_metadata_images_list(
             feature_dict={'aspectOfHand': label})
         for image in label_images_list:
             features[image] = database_features[image]
     return features