MCPcopy Create free account
hub / github.com/DeepGraphLearning/graphvite / train_label_preprocess

Method train_label_preprocess

python/graphvite/dataset.py:979–984  ·  view source on GitHub ↗
(self, save_file)

Source from the content-addressed store, hash-verified

977 return features
978
979 def train_label_preprocess(self, save_file):
980 image_files = glob.glob(os.path.join(self.train_image, "*/*.JPEG"))
981 labels = [os.path.basename(os.path.dirname(image_file)) for image_file in image_files]
982 # be consistent with the order in torch.utils.data.DataLoader
983 labels = sorted(labels)
984 self.readable_label(labels, save_file)
985
986 def train_hierarchical_label_preprocess(self, save_file):
987 image_files = glob.glob(os.path.join(self.train_image, "*/*.JPEG"))

Callers

nothing calls this directly

Calls 1

readable_labelMethod · 0.95

Tested by

no test coverage detected