Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/wolfstefan/fungi2023
https://github.com/wolfstefan/fungi2023
Last synced: 7 days ago
JSON representation
- Host: GitHub
- URL: https://github.com/wolfstefan/fungi2023
- Owner: wolfstefan
- Created: 2023-05-24T22:04:31.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2023-06-07T23:21:55.000Z (over 1 year ago)
- Last Synced: 2024-08-02T15:30:41.438Z (3 months ago)
- Language: Python
- Size: 128 KB
- Stars: 1
- Watchers: 4
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Transformer-based Fine-Grained Fungi Classification in an Open-Set Scenario
This repository is targeted towards solving the FungiCLEF 2023 (https://huggingface.co/spaces/competitions/FungiCLEF2023) challenge. It is based on MMPreTrain (https://github.com/open-mmlab/mmpretrain).
## Usage
### Installation
```bash
conda create -n fungi2023 python=3.10 pytorch=2.0.1 torchvision=0.15.2 pytorch-cuda=11.8 -c pytorch -c nvidia
conda activate fungi2023
pip install -r requirements.txt
mim install "mmpretrain==1.0.0rc7"
```
### Data
The challenge data has to be downloaded and put into _data/fungiclef2022/_.
### Training
```bash
bash tools/dist_train.sh configs/swinv2_base_w24_b32x4-fp16_fungi+val_res_384_cb_epochs_6.py 4
```
### Inference on pre-trained models
```bash
python tools/test_generate_result_pre-consensus_tta.py models/swinv2_base_w24_b32x4-fp16_fungi+val_res_384_cb_epochs_6.py models/swinv2_base_w24_b32x4-fp16_fungi+val_res_384_cb_epochs_6_20230524-a251a50a.pth results.csv --threshold 0.2 --no-scores
```