Skip to content
Snippets Groups Projects
Commit 86802244 authored by blia's avatar blia
Browse files

Update README.md

parent 9c3f7e85
No related branches found
No related tags found
No related merge requests found
......@@ -33,7 +33,7 @@ In order to evaluate the model at each acquisition step, run
``` python
python3 -c 'import Test as te;te.running_test_for_single_acquisition_step(model_dir)'
Args:
model_dir: the directory that saves the model ckpt
model_dir: the directory that saves the model ckpt
```
#### Train the model
......@@ -43,15 +43,16 @@ Args:
```python
python3 -c 'import Train_Active_Full_Im as tafi;tafi.running_loop_active_learning_full_image(stage)'
Args:
stage: int, 0:random, 1:VarRatio, 2:Entropy, 3:BALD
stage: int, 0:random, 1:VarRatio, 2:Entropy, 3:BALD
```
- For region based active learning, run
```python
python3 -c 'import Train_Active_Region_Im as tari;tari.running_loop_active_learning_region(stage)'
Args:
stage: int, 0:random, 1:VarRatio, 2:Entropy, 3:BALD
```
```python
python3 -c 'import Train_Active_Region_Im as tari;tari.running_loop_active_learning_region(stage)'
Args:
stage: int, 0:random, 1:VarRatio, 2:Entropy, 3:BALD
```
#### Reproduce figures
......@@ -64,6 +65,6 @@ wget https:
tar -xzvf calibration_score.tar.gz
cd ..
cd eval_calibration
python3 visualize_calibration_score.py --save False
python3 visualize_calibration_score.py --save True
```
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment