UPD: stage b4 sync

samjmolyneux · Jan 28, 2021 · 56c7937 · 56c7937
1 parent 1da4965
commit 56c7937
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/run_tile.sh b/run_tile.sh
@@ -1,13 +1,15 @@
 python run_infer.py \
---gpu='0,1' \
---nr_types=6 \
+--gpu='1' \
+--nr_types=0 \
 --type_info_path=type_info.json \
 --batch_size=64 \
---model_mode=fast \
---model_path=../pretrained/hovernet_fast_pannuke_type_tf2pytorch..tar \
+--model_mode=original \
+--model_path=../pretrained/hovernet_original_consep_notype_tf2pytorch.tar \
 --nr_inference_workers=8 \
 --nr_post_proc_workers=16 \
 tile \
+--input_dir=dataset/sample_tiles/imgs/ \
+--output_dir=dataset/sample_tiles/pred/ \
 --mem_usage=0.1 \
 --draw_dot \
 --save_qupath