wast-Classification & Localization

-NU-Machine Learning and Pattern Recognition project

Image Classification

--Custom customized loader and model

--WasteSort_xx.py main training function

Object Localization

--main.py main training function

--evaluation.py evluation code

--my_dataset customized dataset loader

--my_loss.py customized loss function

--my_transfrom.py customized data transform style

--my_model customized network model