Skip to content

Latest commit

 

History

History

image_classification_on_custom_data

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
base_model.py
base_model.py
 
 
data_generator.py
data_generator.py
 
 
hyperparameter_search.py
hyperparameter_search.py
 
 
predict.py
predict.py
 
 
trainer.py
trainer.py
 
 
utils.py
utils.py
 
 

README.md

Dependencies

tf-nightly-gpu
sklearn
matplotlib

Dataset structure

├── root_image_dir                # input image files
│   ├── class_1_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_2_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_3_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_4_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_5_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_6_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)
│   ├── class_6_images_dir        # contains image files (jpg/jpeg/png/gif/bmp etc.)