marishmak / GarbageSorter

MIT License
0 stars 0 forks source link

Build a first classification model #1

Closed Post-Modern28 closed 4 weeks ago

Post-Modern28 commented 4 weeks ago

Results of testing pretrained models from torchvision: VGG16 showed a mediocre result of 0.7 accuracy. ResNet18 has shown an accuracy slightly below 0.9. All other models (ResNet50/152, GoogLeNet, EfficientNetV2_S) have shown an accuracy above 0.9.