# -*- coding: UTF-8 -*-


# image shape
IMG_ROW, IMG_COL = 50, 50

# model train
batch_size = 30
num_classes = 2
epochs = 10

# model name
model_name = "model/trained_model_1.h5"

# image path
IMAGE_PATH = "image/"
TRAIN_PATH = "train/"
PREDICT_PATH = "predict/"

# augmentation size for one class
augmentation_size = 320

