def load_model(self):
if self.img_model == 'alexnet':
img_output = img_alexnet_layers(
self.img, self.batch_size, self.output_dim,
self.stage, self.model_weights, val_batch_size=self.val_batch_size)
else:
raise Exception('cannot use such CNN model as ' + self.img_model)
return img_output
def load_model(self):
if self.img_model == 'alexnet':
img_output = img_alexnet_layers(
self.img, self.batch_size, self.output_dim,
self.stage, self.model_weights, val_batch_size=self.val_batch_size)
else:
raise Exception('cannot use such CNN model as ' + self.img_model)
return img_output
def load_model(self):
# print('Just tracking...') # after Ln42
if self.img_model == 'alexnet': # from args
img_output = img_alexnet_layers(self.img, self.batch_size,
self.output_dim, self.stage,
self.model_weights, self.with_tanh,
self.val_batch_size)
else:
raise Exception('cannot use such CNN model as ' + self.img_model)
return img_output
