| # 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 | |
| ```bash | |
| 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 | |
| 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. | |
| ```bash | |
| python script.py | |
| ``` |