xiaoyi-feng / pytorch_flask Goto Github PK
View Code? Open in Web Editor NEWuse pytorch to train the Resnet-18 model based on own image dataset to clasify the trash;and use flask to invoke the saved .pt model to predict the class of the images from Android front end.