Add model V2 based on custom dataset

efficientnet_custom/categories.txt

@@ -0,0 +1,16 @@
+car_battery:0
+cardboard:1
+food_organics:2
+glass:3
+metal:4
+mirror:5
+miscellaneous_trash:6
+neon:7
+paper:8
+pharmacy:9
+plastic:10
+printer_cartridge:11
+textile_trash:12
+tire:13
+vegetation:14
+wood:15

efficientnet_custom/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dad0e3448be385310ea3a94e198b5a488400284ae4dbc93b4fc94eb6aa756248
+size 16412071

efficientnet_custom/training_history.txt

@@ -0,0 +1,50 @@
+Epoch 1: Train Loss: 0.9562614777588075, Train Accuracy: 0.723508064516129, Val Accuracy: 0.8787096774193548
+Epoch 2: Train Loss: 0.39243106489219975, Train Accuracy: 0.8763709677419355, Val Accuracy: 0.9158064516129032
+Epoch 3: Train Loss: 0.27795935158287327, Train Accuracy: 0.9105645161290322, Val Accuracy: 0.9325806451612904
+Epoch 4: Train Loss: 0.20849699167474622, Train Accuracy: 0.9313709677419355, Val Accuracy: 0.9420967741935484
+Epoch 5: Train Loss: 0.16578860777760707, Train Accuracy: 0.9453629032258064, Val Accuracy: 0.945
+Epoch 6: Train Loss: 0.14188145429256463, Train Accuracy: 0.9547983870967742, Val Accuracy: 0.95
+Epoch 7: Train Loss: 0.12035028157935988, Train Accuracy: 0.9605241935483871, Val Accuracy: 0.9509677419354838
+Epoch 8: Train Loss: 0.10934833347617137, Train Accuracy: 0.9649596774193548, Val Accuracy: 0.9535483870967741
+Epoch 9: Train Loss: 0.09444800416608491, Train Accuracy: 0.9691129032258065, Val Accuracy: 0.9537096774193549
+Epoch 10: Train Loss: 0.08487475175590765, Train Accuracy: 0.9709677419354839, Val Accuracy: 0.9535483870967741
+Epoch 11: Train Loss: 0.079901130413937, Train Accuracy: 0.9735887096774194, Val Accuracy: 0.9566129032258065
+Epoch 12: Train Loss: 0.06660182510968297, Train Accuracy: 0.9777016129032258, Val Accuracy: 0.9548387096774194
+Epoch 13: Train Loss: 0.06455958083302023, Train Accuracy: 0.9794354838709678, Val Accuracy: 0.9574193548387097
+Epoch 14: Train Loss: 0.06489166927959529, Train Accuracy: 0.9783064516129032, Val Accuracy: 0.9551612903225807
+Epoch 15: Train Loss: 0.058866560919240356, Train Accuracy: 0.9799193548387096, Val Accuracy: 0.959516129032258
+Epoch 16: Train Loss: 0.06079556260528343, Train Accuracy: 0.9796370967741935, Val Accuracy: 0.9588709677419355
+Epoch 17: Train Loss: 0.0563415158410827, Train Accuracy: 0.9825, Val Accuracy: 0.9556451612903226
+Epoch 18: Train Loss: 0.04873851758638217, Train Accuracy: 0.9841935483870968, Val Accuracy: 0.9604838709677419
+Epoch 19: Train Loss: 0.046843323398892196, Train Accuracy: 0.9854032258064516, Val Accuracy: 0.9554838709677419
+Epoch 20: Train Loss: 0.05091928186826408, Train Accuracy: 0.9841129032258065, Val Accuracy: 0.9564516129032258
+Epoch 21: Train Loss: 0.05007656872054682, Train Accuracy: 0.9829435483870967, Val Accuracy: 0.9570967741935484
+Epoch 22: Train Loss: 0.04347503942249703, Train Accuracy: 0.9860887096774194, Val Accuracy: 0.9579032258064516
+Epoch 23: Train Loss: 0.040597257405714765, Train Accuracy: 0.985524193548387, Val Accuracy: 0.9604838709677419
+Epoch 24: Train Loss: 0.04470969083373465, Train Accuracy: 0.984516129032258, Val Accuracy: 0.9575806451612904
+Epoch 25: Train Loss: 0.03362339327479863, Train Accuracy: 0.9892741935483871, Val Accuracy: 0.9598387096774194
+Epoch 26: Train Loss: 0.028827502301653787, Train Accuracy: 0.990524193548387, Val Accuracy: 0.962258064516129
+Epoch 27: Train Loss: 0.029744031447954775, Train Accuracy: 0.9903629032258064, Val Accuracy: 0.9619354838709677
+Epoch 28: Train Loss: 0.027638619745322952, Train Accuracy: 0.9913306451612903, Val Accuracy: 0.9604838709677419
+Epoch 29: Train Loss: 0.02415655878165947, Train Accuracy: 0.9920161290322581, Val Accuracy: 0.9630645161290322
+Epoch 30: Train Loss: 0.023394590490309883, Train Accuracy: 0.9925806451612903, Val Accuracy: 0.9633870967741935
+Epoch 31: Train Loss: 0.02522998656998689, Train Accuracy: 0.9918951612903226, Val Accuracy: 0.9641935483870968
+Epoch 32: Train Loss: 0.024817155856308677, Train Accuracy: 0.9915725806451613, Val Accuracy: 0.9612903225806452
+Epoch 33: Train Loss: 0.025004426797292374, Train Accuracy: 0.9917338709677419, Val Accuracy: 0.9625806451612903
+Epoch 34: Train Loss: 0.021803508300311684, Train Accuracy: 0.9929435483870968, Val Accuracy: 0.9598387096774194
+Epoch 35: Train Loss: 0.022351764769991858, Train Accuracy: 0.9927016129032258, Val Accuracy: 0.9601612903225807
+Epoch 36: Train Loss: 0.023243420824686428, Train Accuracy: 0.9925403225806452, Val Accuracy: 0.9643548387096774
+Epoch 37: Train Loss: 0.02183254863364592, Train Accuracy: 0.9930241935483871, Val Accuracy: 0.962741935483871
+Epoch 38: Train Loss: 0.02350909288797993, Train Accuracy: 0.9924596774193548, Val Accuracy: 0.9643548387096774
+Epoch 39: Train Loss: 0.023081076407425797, Train Accuracy: 0.9924596774193548, Val Accuracy: 0.9620967741935483
+Epoch 40: Train Loss: 0.021087470210247462, Train Accuracy: 0.9933064516129032, Val Accuracy: 0.9604838709677419
+Epoch 41: Train Loss: 0.021318646008899857, Train Accuracy: 0.9931854838709677, Val Accuracy: 0.9632258064516129
+Epoch 42: Train Loss: 0.021894983218930242, Train Accuracy: 0.9930645161290322, Val Accuracy: 0.9637096774193549
+Epoch 43: Train Loss: 0.017173028700194517, Train Accuracy: 0.9942741935483871, Val Accuracy: 0.9646774193548387
+Epoch 44: Train Loss: 0.017876166943460704, Train Accuracy: 0.9940725806451612, Val Accuracy: 0.9664516129032258
+Epoch 45: Train Loss: 0.017069045818512207, Train Accuracy: 0.9942741935483871, Val Accuracy: 0.9662903225806452
+Epoch 46: Train Loss: 0.015354544402122678, Train Accuracy: 0.9951612903225806, Val Accuracy: 0.9617741935483871
+Epoch 47: Train Loss: 0.01648524471673168, Train Accuracy: 0.9948790322580645, Val Accuracy: 0.9654838709677419
+Epoch 48: Train Loss: 0.01604426196372632, Train Accuracy: 0.9945161290322581, Val Accuracy: 0.9646774193548387
+Epoch 49: Train Loss: 0.015699545858529278, Train Accuracy: 0.9947177419354839, Val Accuracy: 0.9664516129032258
+Epoch 50: Train Loss: 0.015483704387114548, Train Accuracy: 0.9948387096774194, Val Accuracy: 0.9656451612903226