stefanwolf commited on
Commit
c860968
1 Parent(s): a71cb96

Added README

Browse files
Files changed (1) hide show
  1. README.md +30 -0
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
+ ```