added images

README.md

@@ -2,15 +2,22 @@
 This repository contains the `classification` model for the tasks required to be accomplised as a part of the Forst Department Project done by Aerodynamics Club, BITS Goa.
 
 ### Classification model (WIP)
-We performed Transfer Learning on pretrained `Resnet18` and `VGG` models from Pytorch. We use the ensemble of the two for inference. \
+We performed Transfer Learning on pretrained `Resnet18` and `VGG` models from Pytorch. We use the ensemble of the two for inference.
 
 | Model         | Epochs trained | Accuracy | F1 Score |
 | :---          |       :---:    | :---:    | :---:    |
 |`Resnet18`     | 30            |   81.67   | 0.81     |
 |`VGG`          | 10             | 83.165   |   0.81   |
 |`Ensemble`     | -              | 86.334   |   0.86   |
 
-## Heatmap
-<img src="SuhrudhSarathy/ForestDeptProj/fig/heatmap.png" alt="Heatmap after Ensemble" style="height: 100px; width:100px;"/>
+## Confusion Matrix
+![Confusion Matrix](fig/heatmap.png) 
+
+Confusion matrix generated using the ensemble model.
+
+## Sample Results
+The model was tested on real world data collected from our visits to Mangrove forests. More sample results can be found [here]('fig/IMG_0629_test.png')
+![Sample](fig/IMG_0629_test.png)
+![Sample 2](fig/IMG_0630_test.png)
 
 

logs/IMG_0629.log

@@ -0,0 +1,122 @@
+File: IMG_0629.log
+Ensemble: 
+Predicted Class: Kandelia candel
+Classes Distribution:
+1. Acanthus ilicifolius: 11.72835636138916
+2. Acrostichum aureum: 0.13847225904464722
+3. Avicennia alba: 0.01707618124783039
+4. Avicennia marina: 0.7901577353477478
+5. Bruguiera cylindrica: 0.045387499034404755
+6. Bruguiera gymnorrihza: 0.12923742830753326
+7. Ceriops tagal: 13.03290843963623
+8. Derris heterophylla: 1.0745940208435059
+9. Excoecaria agallocha: 0.30272459983825684
+10. Kandelia candel: 33.342899322509766
+11. Lumnizera recemosa: 0.6350750923156738
+12. Rhizophora apiculata: 0.12404817342758179
+13. Rhizophora mucronata: 20.177953720092773
+14. Sonneratia alba: 7.513378620147705
+15. Sonneratia caseolaris: 1.2508544921875
+16. Aegiceras corniculatum: 9.696881294250488
+---
+Resnet: 
+Predicted Class: Kandelia candel
+Classes Distribution:
+1. Acanthus ilicifolius: 20.506837844848633
+2. Acrostichum aureum: 0.181674525141716
+3. Avicennia alba: 0.00018319931405130774
+4. Avicennia marina: 1.4099980592727661
+5. Bruguiera cylindrica: 0.004797529429197311
+6. Bruguiera gymnorrihza: 0.21846042573451996
+7. Ceriops tagal: 0.07689297944307327
+8. Derris heterophylla: 0.009683128446340561
+9. Excoecaria agallocha: 0.42889276146888733
+10. Kandelia candel: 66.67536163330078
+11. Lumnizera recemosa: 1.2070550918579102
+12. Rhizophora apiculata: 0.0027297958731651306
+13. Rhizophora mucronata: 2.323148488998413
+14. Sonneratia alba: 0.04760955274105072
+15. Sonneratia caseolaris: 0.016987401992082596
+16. Aegiceras corniculatum: 6.889695167541504
+---
+VGG: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 2.949873685836792
+2. Acrostichum aureum: 0.09527000039815903
+3. Avicennia alba: 0.03396916016936302
+4. Avicennia marina: 0.17031733691692352
+5. Bruguiera cylindrica: 0.0859774723649025
+6. Bruguiera gymnorrihza: 0.04001442342996597
+7. Ceriops tagal: 25.988924026489258
+8. Derris heterophylla: 2.139504909515381
+9. Excoecaria agallocha: 0.17655645310878754
+10. Kandelia candel: 0.010434117168188095
+11. Lumnizera recemosa: 0.06309517472982407
+12. Rhizophora apiculata: 0.24536654353141785
+13. Rhizophora mucronata: 38.03275680541992
+14. Sonneratia alba: 14.979147911071777
+15. Sonneratia caseolaris: 2.4847216606140137
+16. Aegiceras corniculatum: 12.504068374633789
+---**---
+File: IMG_0629.log
+Ensemble: 
+Predicted Class: Kandelia candel
+Classes Distribution:
+1. Acanthus ilicifolius: 11.72835636138916
+2. Acrostichum aureum: 0.13847225904464722
+3. Avicennia alba: 0.01707618124783039
+4. Avicennia marina: 0.7901577353477478
+5. Bruguiera cylindrica: 0.045387499034404755
+6. Bruguiera gymnorrihza: 0.12923742830753326
+7. Ceriops tagal: 13.03290843963623
+8. Derris heterophylla: 1.0745940208435059
+9. Excoecaria agallocha: 0.30272459983825684
+10. Kandelia candel: 33.342899322509766
+11. Lumnizera recemosa: 0.6350750923156738
+12. Rhizophora apiculata: 0.12404817342758179
+13. Rhizophora mucronata: 20.177953720092773
+14. Sonneratia alba: 7.513378620147705
+15. Sonneratia caseolaris: 1.2508544921875
+16. Aegiceras corniculatum: 9.696881294250488
+---
+Resnet: 
+Predicted Class: Kandelia candel
+Classes Distribution:
+1. Acanthus ilicifolius: 20.506837844848633
+2. Acrostichum aureum: 0.181674525141716
+3. Avicennia alba: 0.00018319931405130774
+4. Avicennia marina: 1.4099980592727661
+5. Bruguiera cylindrica: 0.004797529429197311
+6. Bruguiera gymnorrihza: 0.21846042573451996
+7. Ceriops tagal: 0.07689297944307327
+8. Derris heterophylla: 0.009683128446340561
+9. Excoecaria agallocha: 0.42889276146888733
+10. Kandelia candel: 66.67536163330078
+11. Lumnizera recemosa: 1.2070550918579102
+12. Rhizophora apiculata: 0.0027297958731651306
+13. Rhizophora mucronata: 2.323148488998413
+14. Sonneratia alba: 0.04760955274105072
+15. Sonneratia caseolaris: 0.016987401992082596
+16. Aegiceras corniculatum: 6.889695167541504
+---
+VGG: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 2.949873685836792
+2. Acrostichum aureum: 0.09527000039815903
+3. Avicennia alba: 0.03396916016936302
+4. Avicennia marina: 0.17031733691692352
+5. Bruguiera cylindrica: 0.0859774723649025
+6. Bruguiera gymnorrihza: 0.04001442342996597
+7. Ceriops tagal: 25.988924026489258
+8. Derris heterophylla: 2.139504909515381
+9. Excoecaria agallocha: 0.17655645310878754
+10. Kandelia candel: 0.010434117168188095
+11. Lumnizera recemosa: 0.06309517472982407
+12. Rhizophora apiculata: 0.24536654353141785
+13. Rhizophora mucronata: 38.03275680541992
+14. Sonneratia alba: 14.979147911071777
+15. Sonneratia caseolaris: 2.4847216606140137
+16. Aegiceras corniculatum: 12.504068374633789
+---**---

logs/IMG_0630.log

@@ -0,0 +1,183 @@
+File: IMG_0630.log
+Ensemble: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 5.07304573059082
+2. Acrostichum aureum: 0.37666383385658264
+3. Avicennia alba: 0.04225436598062515
+4. Avicennia marina: 0.15483416616916656
+5. Bruguiera cylindrica: 0.1386629194021225
+6. Bruguiera gymnorrihza: 0.25603124499320984
+7. Ceriops tagal: 27.977783203125
+8. Derris heterophylla: 3.964148759841919
+9. Excoecaria agallocha: 0.6488136053085327
+10. Kandelia candel: 0.38064488768577576
+11. Lumnizera recemosa: 0.7796587347984314
+12. Rhizophora apiculata: 14.52924919128418
+13. Rhizophora mucronata: 28.451871871948242
+14. Sonneratia alba: 2.378612995147705
+15. Sonneratia caseolaris: 6.074620723724365
+16. Aegiceras corniculatum: 8.77310562133789
+---
+Resnet: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 8.241856575012207
+2. Acrostichum aureum: 0.4012977182865143
+3. Avicennia alba: 0.0012743371771648526
+4. Avicennia marina: 0.0334632508456707
+5. Bruguiera cylindrica: 0.08958852291107178
+6. Bruguiera gymnorrihza: 0.04349733516573906
+7. Ceriops tagal: 20.555213928222656
+8. Derris heterophylla: 6.682247161865234
+9. Excoecaria agallocha: 0.3442293107509613
+10. Kandelia candel: 0.18194973468780518
+11. Lumnizera recemosa: 0.27449244260787964
+12. Rhizophora apiculata: 0.6251317262649536
+13. Rhizophora mucronata: 40.358863830566406
+14. Sonneratia alba: 0.1994485706090927
+15. Sonneratia caseolaris: 9.615466117858887
+16. Aegiceras corniculatum: 12.351972579956055
+---
+VGG: 
+Predicted Class: Ceriops tagal
+Classes Distribution:
+1. Acanthus ilicifolius: 1.9042352437973022
+2. Acrostichum aureum: 0.3520299792289734
+3. Avicennia alba: 0.08323439210653305
+4. Avicennia marina: 0.27620506286621094
+5. Bruguiera cylindrica: 0.18773731589317322
+6. Bruguiera gymnorrihza: 0.4685651659965515
+7. Ceriops tagal: 35.400352478027344
+8. Derris heterophylla: 1.2460509538650513
+9. Excoecaria agallocha: 0.9533978700637817
+10. Kandelia candel: 0.5793400406837463
+11. Lumnizera recemosa: 1.284825086593628
+12. Rhizophora apiculata: 28.433366775512695
+13. Rhizophora mucronata: 16.544876098632812
+14. Sonneratia alba: 4.557777404785156
+15. Sonneratia caseolaris: 2.5337753295898438
+16. Aegiceras corniculatum: 5.194238662719727
+---**---
+File: IMG_0630.log
+Ensemble: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 5.07304573059082
+2. Acrostichum aureum: 0.37666383385658264
+3. Avicennia alba: 0.04225436598062515
+4. Avicennia marina: 0.15483416616916656
+5. Bruguiera cylindrica: 0.1386629194021225
+6. Bruguiera gymnorrihza: 0.25603124499320984
+7. Ceriops tagal: 27.977783203125
+8. Derris heterophylla: 3.964148759841919
+9. Excoecaria agallocha: 0.6488136053085327
+10. Kandelia candel: 0.38064488768577576
+11. Lumnizera recemosa: 0.7796587347984314
+12. Rhizophora apiculata: 14.52924919128418
+13. Rhizophora mucronata: 28.451871871948242
+14. Sonneratia alba: 2.378612995147705
+15. Sonneratia caseolaris: 6.074620723724365
+16. Aegiceras corniculatum: 8.77310562133789
+---
+Resnet: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 8.241856575012207
+2. Acrostichum aureum: 0.4012977182865143
+3. Avicennia alba: 0.0012743371771648526
+4. Avicennia marina: 0.0334632508456707
+5. Bruguiera cylindrica: 0.08958852291107178
+6. Bruguiera gymnorrihza: 0.04349733516573906
+7. Ceriops tagal: 20.555213928222656
+8. Derris heterophylla: 6.682247161865234
+9. Excoecaria agallocha: 0.3442293107509613
+10. Kandelia candel: 0.18194973468780518
+11. Lumnizera recemosa: 0.27449244260787964
+12. Rhizophora apiculata: 0.6251317262649536
+13. Rhizophora mucronata: 40.358863830566406
+14. Sonneratia alba: 0.1994485706090927
+15. Sonneratia caseolaris: 9.615466117858887
+16. Aegiceras corniculatum: 12.351972579956055
+---
+VGG: 
+Predicted Class: Ceriops tagal
+Classes Distribution:
+1. Acanthus ilicifolius: 1.9042352437973022
+2. Acrostichum aureum: 0.3520299792289734
+3. Avicennia alba: 0.08323439210653305
+4. Avicennia marina: 0.27620506286621094
+5. Bruguiera cylindrica: 0.18773731589317322
+6. Bruguiera gymnorrihza: 0.4685651659965515
+7. Ceriops tagal: 35.400352478027344
+8. Derris heterophylla: 1.2460509538650513
+9. Excoecaria agallocha: 0.9533978700637817
+10. Kandelia candel: 0.5793400406837463
+11. Lumnizera recemosa: 1.284825086593628
+12. Rhizophora apiculata: 28.433366775512695
+13. Rhizophora mucronata: 16.544876098632812
+14. Sonneratia alba: 4.557777404785156
+15. Sonneratia caseolaris: 2.5337753295898438
+16. Aegiceras corniculatum: 5.194238662719727
+---**---
+File: IMG_0630.log
+Ensemble: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 0.9521187543869019
+2. Acrostichum aureum: 0.1760152280330658
+3. Avicennia alba: 0.041617218405008316
+4. Avicennia marina: 0.1383167803287506
+5. Bruguiera cylindrica: 0.09623689949512482
+6. Bruguiera gymnorrihza: 0.23429575562477112
+7. Ceriops tagal: 17.70103645324707
+8. Derris heterophylla: 0.6230839490890503
+9. Excoecaria agallocha: 0.47863152623176575
+10. Kandelia candel: 0.2896721363067627
+11. Lumnizera recemosa: 0.6452482342720032
+12. Rhizophora apiculata: 14.225202560424805
+13. Rhizophora mucronata: 58.25287628173828
+14. Sonneratia alba: 2.278890371322632
+15. Sonneratia caseolaris: 1.2692240476608276
+16. Aegiceras corniculatum: 2.5975327491760254
+---
+Resnet: 
+Predicted Class: Rhizophora mucronata
+Classes Distribution:
+1. Acanthus ilicifolius: 2.2152639758132864e-06
+2. Acrostichum aureum: 4.940263806929579e-07
+3. Avicennia alba: 4.139608478226364e-08
+4. Avicennia marina: 0.00042849802412092686
+5. Bruguiera cylindrica: 0.0047364793717861176
+6. Bruguiera gymnorrihza: 2.6374518711236306e-05
+7. Ceriops tagal: 0.0017213765531778336
+8. Derris heterophylla: 0.00011699354217853397
+9. Excoecaria agallocha: 0.003865152830258012
+10. Kandelia candel: 4.191454081592383e-06
+11. Lumnizera recemosa: 0.005671343300491571
+12. Rhizophora apiculata: 0.017037907615303993
+13. Rhizophora mucronata: 99.96088409423828
+14. Sonneratia alba: 3.457371349213645e-06
+15. Sonneratia caseolaris: 0.004672559909522533
+16. Aegiceras corniculatum: 0.0008271338883787394
+---
+VGG: 
+Predicted Class: Ceriops tagal
+Classes Distribution:
+1. Acanthus ilicifolius: 1.9042352437973022
+2. Acrostichum aureum: 0.3520299792289734
+3. Avicennia alba: 0.08323439210653305
+4. Avicennia marina: 0.27620506286621094
+5. Bruguiera cylindrica: 0.18773731589317322
+6. Bruguiera gymnorrihza: 0.4685651659965515
+7. Ceriops tagal: 35.400352478027344
+8. Derris heterophylla: 1.2460509538650513
+9. Excoecaria agallocha: 0.9533978700637817
+10. Kandelia candel: 0.5793400406837463
+11. Lumnizera recemosa: 1.284825086593628
+12. Rhizophora apiculata: 28.433366775512695
+13. Rhizophora mucronata: 16.544876098632812
+14. Sonneratia alba: 4.557777404785156
+15. Sonneratia caseolaris: 2.5337753295898438
+16. Aegiceras corniculatum: 5.194238662719727
+---**---