File size: 1,147 Bytes
7318fe0 c860968 7318fe0 c860968 7318fe0 c860968 7318fe0 c860968 7318fe0 c860968 7318fe0 c860968 7318fe0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | # 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
``` |