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