SoulPerforms commited on
Commit
cfc1f5b
·
verified ·
1 Parent(s): 8a3fb56

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -1
README.md CHANGED
@@ -6,4 +6,35 @@ metrics:
6
  pipeline_tag: image-classification
7
  ---
8
  Butterfly image classification model that use pre-trained cnn model resnet18 and fine-tuned the last fully connected layer to classify 75 categories of butterfly species.
9
- The model used the best checkpoint with 90% test accuracy. The model constructed on Pytorch environment
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  pipeline_tag: image-classification
7
  ---
8
  Butterfly image classification model that use pre-trained cnn model resnet18 and fine-tuned the last fully connected layer to classify 75 categories of butterfly species.
9
+ The model used the best checkpoint with 90% test accuracy. The model constructed on Pytorch environment.
10
+
11
+ Epoch: 0 Train Loss: 3.00 Train Accuracy: 0.37 Test Accuracy: 0.70
12
+ Epoch: 1 Train Loss: 1.18 Train Accuracy: 0.76 Test Accuracy: 0.78
13
+ Epoch: 2 Train Loss: 0.75 Train Accuracy: 0.83 Test Accuracy: 0.83
14
+ Epoch: 3 Train Loss: 0.60 Train Accuracy: 0.86 Test Accuracy: 0.85
15
+ Epoch: 4 Train Loss: 0.51 Train Accuracy: 0.88 Test Accuracy: 0.85
16
+ Epoch: 5 Train Loss: 0.43 Train Accuracy: 0.90 Test Accuracy: 0.86
17
+ Epoch: 6 Train Loss: 0.39 Train Accuracy: 0.90 Test Accuracy: 0.87
18
+ Epoch: 7 Train Loss: 0.35 Train Accuracy: 0.91 Test Accuracy: 0.87
19
+ Epoch: 8 Train Loss: 0.32 Train Accuracy: 0.92 Test Accuracy: 0.88
20
+ Epoch: 9 Train Loss: 0.32 Train Accuracy: 0.92 Test Accuracy: 0.89
21
+ Epoch: 10 Train Loss: 0.28 Train Accuracy: 0.93 Test Accuracy: 0.89
22
+ Epoch: 11 Train Loss: 0.28 Train Accuracy: 0.93 Test Accuracy: 0.88
23
+ Epoch: 12 Train Loss: 0.26 Train Accuracy: 0.94 Test Accuracy: 0.89
24
+ Epoch: 13 Train Loss: 0.25 Train Accuracy: 0.94 Test Accuracy: 0.89
25
+ Epoch: 14 Train Loss: 0.23 Train Accuracy: 0.94 Test Accuracy: 0.89
26
+ Epoch: 15 Train Loss: 0.23 Train Accuracy: 0.94 Test Accuracy: 0.89
27
+ Epoch: 16 Train Loss: 0.22 Train Accuracy: 0.94 Test Accuracy: 0.89
28
+ Epoch: 17 Train Loss: 0.21 Train Accuracy: 0.95 Test Accuracy: 0.89
29
+ Epoch: 18 Train Loss: 0.20 Train Accuracy: 0.95 Test Accuracy: 0.90
30
+ Epoch: 19 Train Loss: 0.20 Train Accuracy: 0.95 Test Accuracy: 0.90
31
+ Epoch: 20 Train Loss: 0.20 Train Accuracy: 0.95 Test Accuracy: 0.89
32
+ Epoch: 21 Train Loss: 0.19 Train Accuracy: 0.95 Test Accuracy: 0.90
33
+ Epoch: 22 Train Loss: 0.19 Train Accuracy: 0.95 Test Accuracy: 0.90
34
+ Epoch: 23 Train Loss: 0.19 Train Accuracy: 0.95 Test Accuracy: 0.89
35
+ Epoch: 24 Train Loss: 0.18 Train Accuracy: 0.95 Test Accuracy: 0.90
36
+ ...
37
+ Epoch: 26 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
38
+ Epoch: 27 Train Loss: 0.18 Train Accuracy: 0.95 Test Accuracy: 0.90
39
+ Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
40
+ Epoch: 29 Train Loss: 0.16 Train Accuracy: 0.96 Test Accuracy: 0.89