Parameter Configure
Model Name
-- Select an Option --
Xception
VGG16
VGG19
ResNet50
ResNet101
ResNet50V2
ResNet101V2
ResNet152V2
InceptionV3
MobileNet
MobileNetV2
DenseNet121
DenseNet169
DenseNet201
NASNetMobile
EfficientNetB0
EfficientNetB1
EfficientNetB2
Loss Function
-- Select an Option --
sparse_categorical_crossentropy
categorical_crossentropy
binary_crossentropy
Training data dir
Validation data dir
Image Height
Image Widht
Channels
Agumentation
-- Select an Option --
False
True
Epochs
Classes
Freezing layers
-- Select an Option --
False
True
Number of layers to Freeze
Optimizer
-- Select an Option --
Adam
SGD
RmsProp
Batch Size
Train
Predict