LouisLuu159 / Data-mining-project

Flower Recognition

This Repo is made by Team 5

• Trịnh Quang Minh
• Lưu Quang Tùng
• Viết Minh Hiếu

Project introduction:

• Use CNN to recognize flower
• Use Pytorch
• Use data augmentation
• Use Adam optimizer
• Use StepLR learning rate scheduler

Repo introduction:

• image_loader.py: Get random image from dataset and show transformed and original image
• cnn_main.py: Training data, Validate, show plt comparing train vs validation
• classify_new_image.py: Predict label for given test set