性能#
准确率性能#
准确率性能是通过计算 7 张图像的真实值和推理结果之间的 Dice 系数来衡量的。下表显示了每个类别的 Dice 系数以及每个案例的平均 Dice 系数。
案例 |
图像 |
类别 |
Dice 系数 |
平均 Dice |
|---|---|---|---|---|
1 |
脾脏 |
0.9652 |
0.9652 |
|
2 |
主动脉 |
0.9682 |
0.9682 |
|
3 |
肝脏 |
0.9177 |
0.8008 |
|
肝肿瘤 |
0.6840 |
|||
4 |
肺肿瘤 |
0.8779 |
0.8779 |
|
5 |
结肠癌原发灶 |
0.8421 |
0.8421 |
|
6 |
胃 |
0.9360 |
0.9004 |
|
下腔静脉 |
0.9087 |
|||
胰腺 |
0.7810 |
|||
L1 椎骨 |
0.9793 |
|||
T8 椎骨 |
0.9769 |
|||
大脑 |
0.8203 |
|||
7 |
左侧第 8 肋骨 |
0.8731 |
0.9165 |
|
右侧第 3 肋骨 |
0.9463 |
|||
右侧第 12 肋骨 |
0.9498 |
|||
右侧髂腰肌 |
0.8696 |
|||
心脏 |
0.9439 |
速度性能#
速度性能是通过计算 6 张图像在不同 GPU 上的耗时来衡量的。下表显示了每个案例的 PyTorch 和 TensorRT 模型的結果。
案例 1#
图像: 256cubic.nii.gz
类别 []
尺寸: 256 x 256 x 256
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
3.028 |
2.968 |
TensorRT |
2.702 |
2.710 |
|
A100 |
PyTorch |
7.126 |
7.147 |
TensorRT |
5.856 |
5.674 |
|
RTX 6000 Ada |
PyTorch |
6.757 |
6.763 |
TensorRT |
6.539 |
6.237 |
|
L40s |
PyTorch |
8.767 |
8.823 |
TensorRT |
8.543 |
8.720 |
案例 2#
图像: 256cubic.nii.gz
类别: [脾脏]
尺寸: 256 x 256 x 256
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
2.511 |
2.521 |
TensorRT |
2.287 |
2.298 |
|
A100 |
PyTorch |
6.117 |
6.125 |
TensorRT |
4.0935 |
4.086 |
|
RTX 6000 Ada |
PyTorch |
5.514 |
5.593 |
TensorRT |
5.169 |
5.168 |
|
L40s |
PyTorch |
6.903 |
6.944 |
TensorRT |
6.597 |
6.594 |
案例 3#
图像: 512cubic.nii.gz
类别 []
尺寸: 512 x 512 x 512
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
16.308 |
16.294 |
TensorRT |
15.235 |
15.226 |
|
A100 |
PyTorch |
91.567 |
92.848 |
TensorRT |
87.108 |
84.881 |
|
RTX 6000 Ada |
PyTorch |
109.338 |
109.405 |
TensorRT |
103.284 |
103.386 |
|
L40s |
PyTorch |
96.068 |
97.135 |
TensorRT |
95.064 |
91.982 |
案例 4#
图像: 512cubic.nii.gz
类别: [肝脏]
尺寸: 512 x 512 x 512
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
14.156 |
14.147 |
TensorRT |
13.079 |
13.066 |
|
A100 |
PyTorch |
37.947 |
37.963 |
TensorRT |
34.156 |
33.852 |
|
RTX 6000 Ada |
PyTorch |
32.242 |
32.256 |
TensorRT |
31.434 |
31.149 |
|
L40s |
PyTorch |
33.164 |
34.715 |
TensorRT |
33.244 |
33.151 |
案例 5#
图像: 512-768.nii.gz
类别 []
尺寸: 512 x 512 x 768
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
89.312 |
88.864 |
TensorRT |
87.692 |
87.501 |
|
A100 |
PyTorch |
144.568 |
143.292 |
TensorRT |
140.919 |
140.490 |
|
RTX 6000 Ada |
PyTorch |
160.502 |
160.697 |
TensorRT |
151.359 |
144.844 |
|
L40s |
PyTorch |
146.684 |
150.386 |
TensorRT |
147.603 |
144.717 |
案例 6#
图像: 512-768.nii.gz
类别: [心脏]
尺寸: 512 x 512 x 768
GPU |
模型 |
平均速度 (秒) |
中位速度 (秒) |
|---|---|---|---|
H100 |
PyTorch |
22.957 |
22.991 |
TensorRT |
21.499 |
21.515 |
|
A100 |
PyTorch |
58.676 |
58.831 |
TensorRT |
55.636 |
54.910 |
|
RTX 6000 Ada |
PyTorch |
49.822 |
50.126 |
TensorRT |
33.956 |
33.831 |
|
L40s |
PyTorch |
49.225 |
51.239 |
TensorRT |
48.805 |
49.690 |