def __init__(self,
name='vgg19',
lr=1e-3,
batch_size=BATCH_SIZE,
weights_mode='loss',
optimizer=None):
BaseClassifier.__init__(self, name, IM_SIZE_224, lr, batch_size,
weights_mode, optimizer)
def __init__(self,
name='xception_trainable',
lr=1e-3,
batch_size=BATCH_SIZE,
weights_mode='loss',
optimizer=None):
BaseClassifier.__init__(self, name, IM_SIZE_299, lr, batch_size,
weights_mode, optimizer)
def data_generator(self,
path_image,
train=True,
random_prob=1.0,
**kwargs):
return BaseClassifier.data_generator(self, path_image, train,
random_prob, **kwargs)
def __init__(self, name='inception_resnet_v2', lr=1e-3, batch_size=BATCH_SIZE, weights_mode='loss', optimizer=None):
BaseClassifier.__init__(self, name, IM_SIZE_299,
lr, batch_size, weights_mode, optimizer)
def data_generator(self, path_image, train=True, random_prob=1., **kwargs):
return BaseClassifier.data_generator(self, path_image, train, random_prob, **kwargs)
def __init__(self, name='small', lr=1e-3, batch_size=BATCH_SIZE, weights_mode='loss', optimizer=None):
BaseClassifier.__init__(self,name, IM_SIZE_224,
lr, batch_size, weights_mode, optimizer)