Downloads · 30 days
0
fungi2024
stefanwolf/fungi2024
fungi2024 is a machine learning model from stefanwolf. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository is targeted towards solving the FungiCLEF 2024 (https://huggingface.co/spaces/BVRA/FungiCLEF2024) challenge. It is based on MMPreTrain (https://github.com/open-mmlab/mmpretrain).
Access
Public
Updated Jul 1, 2024
Repo size
2.8 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth414 MB · 99%
At a glance
- Access
- Public
- Created
- May 14, 2024
- Updated
- Jul 1, 2024
- SHA
- 7318fe01
Model card
From the Hugging Face model README
Poison-Aware Open-Set Fungi Classification: Reducing the Risk of Poisonous Confusion
This repository is targeted towards solving the FungiCLEF 2024 (https://huggingface.co/spaces/BVRA/FungiCLEF2024) challenge. It is based on MMPreTrain (https://github.com/open-mmlab/mmpretrain).
Usage
Installation
conda create -p .conda python=3.10 pytorch=2.3 torchvision pytorch-cuda=12.1 -c pytorch -c nvidia
conda activate .conda/
pip install future==1.0.0 tensorboard==2.16.2 pandas==2.2.2
pip install mmpretrain==1.2.0 mmengine==0.10.4 mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.3/index.html
Data
The challenge data has to be downloaded and put into data/fungi2024/.
Training
bash tools/dist_train.sh configs/fungi2023/swinv2_base_w24_b16x8-fp16_fungi+val2_res_384_genus-loss_no-unknown-target-zero.py 8
bash tools/dist_train.sh configs/fungi2023/swinv2_base_w24_b16x8-fp16_fungi+val2_res_384_genus-loss_no-unknown-target-zero_metadata_epochs_4.py 8
Inference
The script script.py performs inference and creates a result file for submission.
python script.py
- Created
- May 14, 2024
- Updated
- Jul 1, 2024