def _extract_feature(self):
with t.no_grad():
self.photo_net.eval()
self.sketch_net.eval()
extractor = Extractor(e_model=self.photo_net,
vis=False,
dataloader=True)
photo_data = extractor.extract(self.photo_test)
extractor.reload_model(self.sketch_net)
sketch_data = extractor.extract(self.sketch_test)
photo_name = photo_data['name']
photo_feature = photo_data['feature']
sketch_name = sketch_data['name']
sketch_feature = sketch_data['feature']
return photo_name, photo_feature, sketch_name, sketch_feature
def _extract_feature_embedding(self):
with t.no_grad():
self.photo_net.eval()
self.sketch_net.eval()
extractor = Extractor(e_model=self.photo_net,
cat_info=False,
vis=False,
dataloader=True)
photo_data = extractor.extract(self.photo_test,
batch_size=self.test_bs)
extractor.reload_model(self.sketch_net)
sketch_data = extractor.extract(self.sketch_test,
batch_size=self.test_bs)
photo_name = photo_data['name']
photo_feature = photo_data['feature']
sketch_name = sketch_data['name']
sketch_feature = sketch_data['feature']
return photo_name, photo_feature, sketch_name, sketch_feature