Data Link -- https://www.kaggle.com/datasets/puneet6060/intel-image-classification?select=seg_train python3 train.py --> train python3 inference.py --> test inference