Skip to content

Commit ed3d194

Browse files
committed
update
Signed-off-by: dongyang0122 <[email protected]>
1 parent 0413df2 commit ed3d194

File tree

1 file changed

+1
-14
lines changed

1 file changed

+1
-14
lines changed

generation/maisi/scripts/compute_fid2p5d.py renamed to generation/maisi/scripts/compute_fid2p5d_ct.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6
2323
NUM_GPUS=7
2424
25-
torchrun --nproc_per_node=${NUM_GPUS} compute_fid2d_mgpu.py \
25+
torchrun --nproc_per_node=${NUM_GPUS} compute_fid2p5d_ct.py \
2626
--model_name "radimagenet_resnet50" \
2727
--data0_dataroot "path/to/datasetA" \
2828
--data0_filelist "path/to/filelistA.txt" \
@@ -64,19 +64,6 @@
6464
output_root (str): Folder where extracted feature .pt files and logs are saved.
6565
target_shape (str): "XxYxZ" shape to which images are padded/cropped/resampled.
6666
67-
Example:
68-
--------
69-
python compute_fid2p5d.py --model_name=radimagenet_resnet50 \\
70-
--data0_dataroot=/data/real_images \\
71-
--data0_filelist=/data/real_list.txt \\
72-
--data0_folder=real \\
73-
--data1_dataroot=/data/synth_images \\
74-
--data1_filelist=/data/synth_list.txt \\
75-
--data1_folder=synth \\
76-
--enable_center_slices=True \\
77-
--enable_padding=True \\
78-
--target_shape=512x512x512
79-
8067
"""
8168

8269
from __future__ import annotations

0 commit comments

Comments
 (0)