stefanwolf
commited on
Commit
•
c860968
1
Parent(s):
a71cb96
Added README
Browse files
README.md
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Transformer-based Fine-Grained Fungi Classification in an Open-Set Scenario
|
2 |
+
|
3 |
+
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).
|
4 |
+
|
5 |
+
## Usage
|
6 |
+
|
7 |
+
### Installation
|
8 |
+
|
9 |
+
```bash
|
10 |
+
conda create -n fungi2023 python=3.10 pytorch=2.0.1 torchvision=0.15.2 pytorch-cuda=11.8 -c pytorch -c nvidia
|
11 |
+
conda activate fungi2023
|
12 |
+
pip install -r requirements.txt
|
13 |
+
mim install "mmpretrain==1.0.0rc7"
|
14 |
+
```
|
15 |
+
|
16 |
+
### Data
|
17 |
+
|
18 |
+
The challenge data has to be downloaded and put into _data/fungiclef2022/_.
|
19 |
+
|
20 |
+
### Training
|
21 |
+
|
22 |
+
```bash
|
23 |
+
bash tools/dist_train.sh configs/swinv2_base_w24_b32x4-fp16_fungi+val_res_384_cb_epochs_6.py 4
|
24 |
+
```
|
25 |
+
|
26 |
+
### Inference on pre-trained models
|
27 |
+
|
28 |
+
```bash
|
29 |
+
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
|
30 |
+
```
|