Update README.md
Browse files
README.md
CHANGED
|
@@ -39,20 +39,20 @@ Please follow the [instruction](https://github.com/yangzhou12/BenchX/blob/releas
|
|
| 39 |
To fine-tune ConVIRT for classification, run this command:
|
| 40 |
|
| 41 |
```
|
| 42 |
-
python bin/train.py config/classification/<dataset_name>/
|
| 43 |
```
|
| 44 |
|
| 45 |
### 2. Segmentation
|
| 46 |
To fine-tune ConVIRT for segmentation, run this command:
|
| 47 |
|
| 48 |
```
|
| 49 |
-
python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/
|
| 50 |
```
|
| 51 |
|
| 52 |
### 3. Report Generation
|
| 53 |
To fine-tune ConVIRT for report generation, run this command:
|
| 54 |
```
|
| 55 |
-
python bin/train.py config/report_generation/<dataset_name>/
|
| 56 |
```
|
| 57 |
|
| 58 |
### 4. Evaluation
|
|
@@ -60,10 +60,10 @@ To evaluate fine-tuned ConVIRT models, run:
|
|
| 60 |
|
| 61 |
```
|
| 62 |
# For classification and report generation
|
| 63 |
-
python bin/test.py config/<task_name>/<dataset_name>/
|
| 64 |
|
| 65 |
# For segmentation
|
| 66 |
-
python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/
|
| 67 |
```
|
| 68 |
|
| 69 |
## Citations
|
|
|
|
| 39 |
To fine-tune ConVIRT for classification, run this command:
|
| 40 |
|
| 41 |
```
|
| 42 |
+
python bin/train.py config/classification/<dataset_name>/convirt.yml
|
| 43 |
```
|
| 44 |
|
| 45 |
### 2. Segmentation
|
| 46 |
To fine-tune ConVIRT for segmentation, run this command:
|
| 47 |
|
| 48 |
```
|
| 49 |
+
python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/convirt.yml
|
| 50 |
```
|
| 51 |
|
| 52 |
### 3. Report Generation
|
| 53 |
To fine-tune ConVIRT for report generation, run this command:
|
| 54 |
```
|
| 55 |
+
python bin/train.py config/report_generation/<dataset_name>/convirt.yml
|
| 56 |
```
|
| 57 |
|
| 58 |
### 4. Evaluation
|
|
|
|
| 60 |
|
| 61 |
```
|
| 62 |
# For classification and report generation
|
| 63 |
+
python bin/test.py config/<task_name>/<dataset_name>/convirt.yml validator.splits=[test] ckpt_dir=<path_to_checkpoint>
|
| 64 |
|
| 65 |
# For segmentation
|
| 66 |
+
python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/convirt.yml <path_to_checkpoint>
|
| 67 |
```
|
| 68 |
|
| 69 |
## Citations
|