Replace link to GitHub with Code Ocean, add new paper citation
#5
by AndiLindner - opened
README.md
CHANGED
|
@@ -34,7 +34,7 @@ The second model version ([2371ff8](https://huggingface.co/RikeB/MaxViT_butterfl
|
|
| 34 |
|
| 35 |
Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library.
|
| 36 |
More information on model training can be found in the publications below.
|
| 37 |
-
Scripts are availble on [
|
| 38 |
|
| 39 |
### Training Data
|
| 40 |
|
|
@@ -47,7 +47,7 @@ Species with less than 50 images were excluded from training. The final dataset
|
|
| 47 |
The first model version was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
|
| 48 |
|
| 49 |
```bibtex
|
| 50 |
-
@Article{
|
| 51 |
author={Barkmann, Friederike
|
| 52 |
and Lindner, Andreas
|
| 53 |
and W{\"u}rflinger, Ronald
|
|
@@ -68,4 +68,17 @@ url={https://doi.org/10.1038/s41597-025-05708-z}
|
|
| 68 |
}
|
| 69 |
```
|
| 70 |
|
| 71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library.
|
| 36 |
More information on model training can be found in the publications below.
|
| 37 |
+
Scripts are availble on [Code Ocean](https://doi.org/10.24433/CO.7366644.v1).
|
| 38 |
|
| 39 |
### Training Data
|
| 40 |
|
|
|
|
| 47 |
The first model version was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
|
| 48 |
|
| 49 |
```bibtex
|
| 50 |
+
@Article{Barkmannetal2025,
|
| 51 |
author={Barkmann, Friederike
|
| 52 |
and Lindner, Andreas
|
| 53 |
and W{\"u}rflinger, Ronald
|
|
|
|
| 68 |
}
|
| 69 |
```
|
| 70 |
|
| 71 |
+
The second model was obtained by a more thorough training described in the following paper:
|
| 72 |
+
|
| 73 |
+
```bibtex
|
| 74 |
+
@article{Barkmannetal2026,
|
| 75 |
+
author = {Barkmann, Friederike and Lindner, Andreas and R{\"u}disser, Johannes},
|
| 76 |
+
title = {Comparing deep learning models for butterfly and moth ({Lepidoptera}) species identification},
|
| 77 |
+
journal = {Insect Conservation and Diversity},
|
| 78 |
+
year = {2026},
|
| 79 |
+
issn = {1752-458X},
|
| 80 |
+
doi = {10.1111/icad.70123},
|
| 81 |
+
url = {https://doi.org/10.1111/icad.70123},
|
| 82 |
+
abstract = {The number of image-based species records has increased exponentially in recent years. Deep learning models offer opportunities for efficient data processing, especially for species identification. Evaluating models on comprehensive datasets can provide insights into their suitability for species identification. Based on a high-quality dataset with more than 500,000 images, we compared 40 models from 12 classes on the classification of 162 butterfly and moth species, and further analysed the results of the best model. The dataset covers most Austrian butterfly species, but a few difficult-to-identify species are not included or were grouped. Multiple models reached validation accuracies >97\%, with larger models generally performing better, but there were some notable exceptions. The MaxViT-tiny model achieved the highest accuracy despite being one of the smaller models. After further hyperparameter optimisation, the performance of the MaxViT was increased to a Top-1 accuracy of 98.14\% and a Top-5 accuracy of 99.53\% on test data. Although methods to correct for class imbalance were applied, recall and precision were generally lower for species with fewer images in the dataset. Our results emphasise the high potential of deep learning for species identification for the species analysed in this study. The results aid projects with limited computational capacities in choosing an efficient and appropriate model. Additional data processing steps, such as applying quality criteria to exclude model predictions with low confidence and expert assessments for difficult-to-determine species, can substantially increase the quality of identification.}
|
| 83 |
+
}
|
| 84 |
+
```
|