Skip to content

Commit cd5e398

Browse files
Can-ZhaoCan-Zhao
committed
add time cost and gpu memory to readme asn add corresponding configs
Signed-off-by: Can-Zhao <[email protected]>
1 parent 2db8c62 commit cd5e398

12 files changed

+242
-5
lines changed

‎generation/maisi/configs/config_infer_16g_256x256x128.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
256,
10+
256,
11+
128
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
1.5,
17+
1.5,
18+
4.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [96,96,96],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 2
23+
}

‎generation/maisi/configs/config_infer_16g_256x256x256.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
256,
10+
256,
11+
256
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
1.5,
17+
1.5,
18+
2.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [48,48,64],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 2
23+
}

‎generation/maisi/configs/config_infer_16g_512x512x128.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
128
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
4.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [64,64,32],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 1
23+
}

‎generation/maisi/configs/config_infer_24g_256x256x256.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
256,
10+
256,
11+
256
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
1.5,
17+
1.5,
18+
2.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [64,64,64],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 4
23+
}

‎generation/maisi/configs/config_infer_24g_512x512x128.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
128
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
4.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [80,80,32],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 1
23+
}

‎generation/maisi/configs/config_infer_24g_512x512x512.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
512
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
1.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [64,64,48],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 2
23+
}

‎generation/maisi/configs/config_infer_32g_512x512x512.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
512
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
1.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [64,64,64],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 2
23+
}

‎generation/maisi/configs/config_infer_80g_512x512x128.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
128
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
4.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [128,128,32],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 4
23+
}

‎generation/maisi/configs/config_infer_80g_512x512x512.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
512
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
1.0
19+
],
20+
"autoencoder_sliding_window_infer_size": [80,80,80],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 2
23+
}

‎generation/maisi/configs/config_infer_80g_512x512x768.json‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
{
2+
"num_output_samples": 1,
3+
"body_region": ["abdomen"],
4+
"anatomy_list": ["liver","hepatic tumor"],
5+
"controllable_anatomy_size": [],
6+
"num_inference_steps": 1000,
7+
"mask_generation_num_inference_steps": 1000,
8+
"output_size": [
9+
512,
10+
512,
11+
768
12+
],
13+
"image_output_ext": ".nii.gz",
14+
"label_output_ext": ".nii.gz",
15+
"spacing": [
16+
0.75,
17+
0.75,
18+
0.66667
19+
],
20+
"autoencoder_sliding_window_infer_size": [80,80,112],
21+
"autoencoder_sliding_window_infer_overlap": 0.25,
22+
"autoencoder_tp_num_splits": 4
23+
}

0 commit comments

Comments
 (0)