Update model with improved regularization and data augmentation

README.md

@@ -30,7 +30,7 @@ This model was trained in Google Colab using a T4 GPU and tracked with MLflow.
 - Bean Rust
 - Angular Leaf Spot
 
-**Validation Accuracy:** 0.9173
+**Validation Accuracy:** 0.8797
 
 ## Training Configuration
 

per_class_metrics.csv

@@ -1,7 +1,7 @@
 ,precision,recall,f1-score,support
-Healthy,1.0,0.7954545454545454,0.8860759493670886,44.0
-Bean Rust,0.8035714285714286,1.0,0.8910891089108911,45.0
-Angular Leaf Spot,1.0,0.9545454545454546,0.9767441860465116,44.0
-accuracy,0.9172932330827067,0.9172932330827067,0.9172932330827067,0.9172932330827067
-macro avg,0.9345238095238096,0.9166666666666666,0.9179697481081638,133.0
-weighted avg,0.9335392051557464,0.9172932330827067,0.9177676380390113,133.0
+Healthy,1.0,0.7272727272727273,0.8421052631578947,44.0
+Bean Rust,0.7377049180327869,1.0,0.8490566037735849,45.0
+Angular Leaf Spot,1.0,0.9090909090909091,0.9523809523809523,44.0
+accuracy,0.8796992481203008,0.8796992481203008,0.8796992481203008,0.8796992481203008
+macro avg,0.912568306010929,0.8787878787878788,0.8811809397708107,133.0
+weighted avg,0.9112535436953039,0.8796992481203008,0.8809394034099294,133.0

resnet18_beans.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f2ba2b958dcf443e818257191c69f9cc804dc0df26934064e371c0f51fb4b5d2
+oid sha256:72a98b3414f9910436d78cf79eefbe4028c8e77bb90d54248e60e51469d0829c
 size 44792395