Spaces:
Sleeping
Sleeping
Saket Shirsath
commited on
Commit
·
63364be
1
Parent(s):
01bb34f
Update index.md
Browse files
index.md
CHANGED
@@ -39,25 +39,25 @@ We started with a single-layer CNN, altering input normalization resolution to s
|
|
39 |
|
40 |
| 25x25 | 50x50 | 100x100 | 150x150 |
|
41 |
| ----- | ----- | ------- | ------- |
|
42 |
-
| <img src="assets/cnn1/cnn1_25_acc.png" height="
|
43 |
-
| <img src="assets/cnn1/cnn1_25_loss.png" height="
|
44 |
|
45 |
The results for the three-layer CNN are worse than the single-layer design. Typically, more layers result in higher validation accuracy, but our dataset is so small and our model is so overfitted that we get the opposite effect, resulting in significantly less validation accuracy across the board.
|
46 |
|
47 |
-
|
|
48 |
-
|
|
49 |
-
|
|
50 |
-
|
|
51 |
-
|
52 |
|
53 |
We also see a similar trend with our four-layer implementation.
|
54 |
|
55 |
|
56 |
-
|
|
57 |
-
|
|
58 |
-
|
|
59 |
-
|
|
60 |
-
|
61 |
|
62 |
Next Steps
|
63 |
Our current dataset is severely limiting the performance of our CNN. There are two ways to expand it: (1) data augmentation and (2) just collecting more images. We will experiment with both of these for the final implementation to get our validation accuracy closer to 100%. It may be possible to improve accuracy by running our dataset images with their associated HPE outputs.
|
|
|
39 |
|
40 |
| 25x25 | 50x50 | 100x100 | 150x150 |
|
41 |
| ----- | ----- | ------- | ------- |
|
42 |
+
| <img src="assets/cnn1/cnn1_25_acc.png" height="150px"> | <img src="assets/cnn1/cnn1_50_acc.png" height="150px"> | <img src="assets/cnn1/cnn1_100_acc.png" height="150px"> | <img src="assets/cnn1/cnn1_150_acc.png" height="150px"> |
|
43 |
+
| <img src="assets/cnn1/cnn1_25_loss.png" height="150px"> | <img src="assets/cnn1/cnn1_50_loss.png" height="150px"> | <img src="assets/cnn1/cnn1_100_loss.png" height="150px"> | <img src="assets/cnn1/cnn1_150_loss.png" height="150px"> |
|
44 |
|
45 |
The results for the three-layer CNN are worse than the single-layer design. Typically, more layers result in higher validation accuracy, but our dataset is so small and our model is so overfitted that we get the opposite effect, resulting in significantly less validation accuracy across the board.
|
46 |
|
47 |
+
| 25x25 | 50x50 | 100x100 | 150x150 |
|
48 |
+
| ----- | ----- | ------- | ------- |
|
49 |
+
| <img src="assets/cnn2/cnn2_25_acc.png" height="100px"> | <img src="assets/cnn2/cnn2_50_acc.png" height="100px"> | <img src="assets/cnn2/cnn2_100_acc.png" height="100px"> | <img src="assets/cnn2/cnn2_150_acc.png" height="100px"> |
|
50 |
+
| <img src="assets/cnn2/cnn2_25_loss.png" height="100px"> | <img src="assets/cnn2/cnn2_50_loss.png" height="100px"> | <img src="assets/cnn2/cnn2_100_loss.png" height="100px"> | <img src="assets/cnn2/cnn2_150_loss.png" height="100px"> |
|
51 |
+
|
52 |
|
53 |
We also see a similar trend with our four-layer implementation.
|
54 |
|
55 |
|
56 |
+
| 25x25 | 50x50 | 100x100 | 150x150 |
|
57 |
+
| ----- | ----- | ------- | ------- |
|
58 |
+
| <img src="assets/cnn3/cnn3_25_acc.png" height="100px"> | <img src="assets/cnn3/cnn3_50_acc.png" height="100px"> | <img src="assets/cnn3/cnn3_100_acc.png" height="100px"> | <img src="assets/cnn3/cnn3_150_acc.png" height="100px"> |
|
59 |
+
| <img src="assets/cnn3/cnn3_25_loss.png" height="100px"> | <img src="assets/cnn3/cnn3_50_loss.png" height="100px"> | <img src="assets/cnn3/cnn3_100_loss.png" height="100px"> | <img src="assets/cnn3/cnn3_150_loss.png" height="100px"> |
|
60 |
+
|
61 |
|
62 |
Next Steps
|
63 |
Our current dataset is severely limiting the performance of our CNN. There are two ways to expand it: (1) data augmentation and (2) just collecting more images. We will experiment with both of these for the final implementation to get our validation accuracy closer to 100%. It may be possible to improve accuracy by running our dataset images with their associated HPE outputs.
|