从您当前提供的测试用例，以及predicted_probabilities 的打印情况来看，数据并未在dcu 上，若想使用dcu 可通过.cuda() 或 .to(device="cuda") 将数据copy 到dcu 使用
0
1
0
0
dcu对于四维tensor,使用argmax(0)无输出，运行nnunet时出错
2024-07-15 00:05:58
95
在环境中做了测试能够输出，但很奇怪代码中不行，
import torch
tensor_test = torch.rand([3,197,221,291])
arg = tensor_test.argmax(0)
max = arg.max()
max
tensor(2)
代码中运行如下语句后segmentation为0，没有得到期望的输出
segmentation = predicted_probabilities.argmax(0)
以下是打印的predicted_probabilities,情况,
predicted_probabilities:torch.Size([3, 197, 221, 291])
predicted_probabilities:tensor([[[[9.9875e-01, 9.9876e-01, 9.9881e-01, ..., 9.9977e-01,
9.9976e-01, 9.9939e-01],
[9.9876e-01, 9.9906e-01, 9.9961e-01, ..., 9.9995e-01,
9.9984e-01, 9.9976e-01],
[9.9881e-01, 9.9959e-01, 9.9975e-01, ..., 9.9998e-01,
9.9995e-01, 9.9983e-01],
...,
[9.9965e-01, 9.9995e-01, 9.9998e-01, ..., 9.9994e-01,
9.9983e-01, 9.9976e-01],
[9.9958e-01, 9.9987e-01, 9.9995e-01, ..., 9.9983e-01,
9.9976e-01, 9.9944e-01],
[9.9887e-01, 9.9962e-01, 9.9987e-01, ..., 9.9976e-01,
9.9944e-01, 9.9884e-01]],
[[9.9876e-01, 9.9897e-01, 9.9956e-01, ..., 9.9994e-01,
9.9981e-01, 9.9977e-01],
[9.9906e-01, 9.9964e-01, 9.9984e-01, ..., 9.9999e-01,
9.9996e-01, 9.9994e-01],
[9.9975e-01, 9.9977e-01, 9.9985e-01, ..., 1.0000e+00,
9.9999e-01, 9.9994e-01],
...,
[9.9994e-01, 9.9998e-01, 9.9999e-01, ..., 9.9999e-01,
9.9995e-01, 9.9982e-01],
[9.9994e-01, 9.9996e-01, 9.9998e-01, ..., 9.9995e-01,
9.9984e-01, 9.9976e-01],
[9.9962e-01, 9.9994e-01, 9.9994e-01, ..., 9.9982e-01,
9.9976e-01, 9.9894e-01]],
[[9.9975e-01, 9.9978e-01, 9.9987e-01, ..., 9.9994e-01,
9.9994e-01, 9.9979e-01],
[9.9977e-01, 9.9994e-01, 9.9994e-01, ..., 9.9999e-01,
9.9998e-01, 9.9994e-01],
[9.9977e-01, 9.9993e-01, 9.9995e-01, ..., 1.0000e+00,
9.9999e-01, 9.9996e-01],
...,
[9.9996e-01, 9.9999e-01, 1.0000e+00, ..., 9.9999e-01,
9.9998e-01, 9.9994e-01],
[9.9994e-01, 9.9999e-01, 1.0000e+00, ..., 9.9999e-01,
9.9996e-01, 9.9981e-01],
[9.9985e-01, 9.9995e-01, 9.9998e-01, ..., 9.9994e-01,
9.9981e-01, 9.9975e-01]],
...,
[[9.9975e-01, 9.9980e-01, 9.9992e-01, ..., 9.9999e-01,
9.9999e-01, 9.9992e-01],
[9.9977e-01, 9.9994e-01, 9.9995e-01, ..., 1.0000e+00,
1.0000e+00, 9.9999e-01],
[9.9982e-01, 9.9995e-01, 9.9998e-01, ..., 1.0000e+00,
1.0000e+00, 9.9999e-01],
...,
[9.9962e-01, 9.9967e-01, 9.9986e-01, ..., 1.0000e+00,
1.0000e+00, 9.9999e-01],
[9.9962e-01, 9.9963e-01, 9.9967e-01, ..., 1.0000e+00,
9.9999e-01, 9.9997e-01],
[9.9888e-01, 9.9956e-01, 9.9962e-01, ..., 9.9999e-01,
9.9999e-01, 9.9992e-01]],
[[9.9939e-01, 9.9976e-01, 9.9977e-01, ..., 9.9999e-01,
9.9996e-01, 9.9979e-01],
[9.9975e-01, 9.9981e-01, 9.9994e-01, ..., 1.0000e+00,
9.9999e-01, 9.9997e-01],
[9.9976e-01, 9.9988e-01, 9.9994e-01, ..., 1.0000e+00,
1.0000e+00, 9.9999e-01],
...,
[9.9901e-01, 9.9962e-01, 9.9962e-01, ..., 1.0000e+00,
9.9999e-01, 9.9994e-01],
[9.9886e-01, 9.9962e-01, 9.9962e-01, ..., 9.9999e-01,
9.9999e-01, 9.9994e-01],
[9.9876e-01, 9.9883e-01, 9.9912e-01, ..., 9.9999e-01,
9.9994e-01, 9.9978e-01]],
[[9.9884e-01, 9.9944e-01, 9.9975e-01, ..., 9.9992e-01,
9.9980e-01, 9.9975e-01],
[9.9944e-01, 9.9976e-01, 9.9977e-01, ..., 9.9999e-01,
9.9996e-01, 9.9981e-01],
[9.9975e-01, 9.9978e-01, 9.9987e-01, ..., 9.9999e-01,
9.9999e-01, 9.9994e-01],
...,
[9.9877e-01, 9.9900e-01, 9.9932e-01, ..., 9.9999e-01,
9.9996e-01, 9.9994e-01],
[9.9876e-01, 9.9880e-01, 9.9889e-01, ..., 9.9999e-01,
9.9995e-01, 9.9981e-01],
[9.9875e-01, 9.9875e-01, 9.9876e-01, ..., 9.9994e-01,
9.9981e-01, 9.9975e-01]]],
[[[9.1075e-04, 9.0367e-04, 8.6909e-04, ..., 1.1234e-04,
1.2147e-04, 3.0526e-04],
[9.0367e-04, 6.8366e-04, 2.9077e-04, ..., 1.4510e-05,
7.1134e-05, 1.2147e-04],
[8.6909e-04, 2.9020e-04, 1.2338e-04, ..., 4.8414e-06,
1.3954e-05, 7.1412e-05],
...,
[4.2801e-05, 1.0209e-05, 2.2603e-06, ..., 2.9772e-05,
8.4144e-05, 1.2194e-04],
[1.0187e-04, 3.3472e-05, 1.0249e-05, ..., 8.4144e-05,
1.2099e-04, 2.7904e-04],
[3.0450e-04, 1.0167e-04, 3.3999e-05, ..., 1.2194e-04,
2.7904e-04, 3.3377e-04]],
[[9.0720e-04, 7.5225e-04, 3.1996e-04, ..., 1.5936e-05,
8.4637e-05, 1.1234e-04],
[6.5747e-04, 2.5711e-04, 1.1913e-04, ..., 4.6378e-06,
1.0910e-05, 1.6442e-05],
[1.2338e-04, 1.2338e-04, 1.0270e-04, ..., 8.3153e-07,
1.9638e-06, 1.5206e-05],
...,
[1.5206e-05, 1.8959e-06, 5.4533e-07, ..., 6.1441e-06,
1.4453e-05, 8.9395e-05],
[1.6442e-05, 9.6281e-06, 1.8959e-06, ..., 1.4510e-05,
7.1134e-05, 1.2147e-04],
[1.0227e-04, 1.6442e-05, 1.5206e-05, ..., 8.9220e-05,
1.2147e-04, 3.0512e-04]],
[[1.2290e-04, 1.1773e-04, 9.3507e-05, ..., 1.5029e-05,
1.5812e-05, 1.0229e-04],
[1.1773e-04, 4.5397e-05, 4.3829e-05, ..., 1.5354e-06,
4.6017e-06, 1.3845e-05],
[1.1773e-04, 4.5397e-05, 3.3671e-05, ..., 4.3139e-07,
1.4881e-06, 4.6196e-06],
...,
[9.0802e-06, 5.8734e-07, 2.8962e-07, ..., 6.1441e-06,
6.1441e-06, 1.5936e-05],
[1.5088e-05, 1.6995e-06, 7.2244e-07, ..., 6.1441e-06,
1.2805e-05, 8.4802e-05],
[4.1170e-05, 1.3160e-05, 4.4427e-06, ..., 1.5936e-05,
8.4637e-05, 1.2266e-04]],
...,
[[1.2266e-04, 9.7988e-05, 3.4401e-05, ..., 4.2559e-06,
5.7719e-06, 3.7635e-05],
[1.1212e-04, 1.6378e-05, 1.4009e-05, ..., 2.5659e-07,
1.2980e-06, 5.7719e-06],
[8.9395e-05, 1.4453e-05, 4.8225e-06, ..., 7.3802e-08,
2.5659e-07, 4.2559e-06],
...,
[4.5383e-05, 3.9277e-05, 1.6699e-05, ..., 3.0590e-07,
7.2244e-07, 5.5943e-06],
[4.5383e-05, 4.4505e-05, 3.9431e-05, ..., 1.3871e-06,
3.9981e-06, 1.3685e-05],
[2.9113e-04, 1.0247e-04, 4.5383e-05, ..., 5.7945e-06,
6.0963e-06, 3.9441e-05]],
[[3.0526e-04, 1.2147e-04, 1.1234e-04, ..., 5.7945e-06,
3.1141e-05, 1.0720e-04],
[1.2242e-04, 8.1873e-05, 1.6701e-05, ..., 1.3871e-06,
3.9981e-06, 1.3685e-05],
[1.1613e-04, 3.2064e-05, 1.6701e-05, ..., 3.0590e-07,
7.2244e-07, 5.5943e-06],
...,
[1.1764e-04, 4.5383e-05, 4.5383e-05, ..., 8.3153e-07,
3.7705e-06, 1.5750e-05],
[2.6611e-04, 4.5383e-05, 4.5383e-05, ..., 3.7559e-06,
5.8857e-06, 1.5936e-05],
[3.3309e-04, 3.0508e-04, 1.9247e-04, ..., 6.0963e-06,
3.1384e-05, 1.0720e-04]],
[[8.2608e-04, 2.7904e-04, 1.2338e-04, ..., 4.1334e-05,
9.7225e-05, 1.2266e-04],
[2.7904e-04, 1.2076e-04, 1.0699e-04, ..., 5.7494e-06,
1.2805e-05, 8.4802e-05],
[1.2338e-04, 1.0309e-04, 3.5770e-05, ..., 2.6426e-06,
5.7494e-06, 1.5936e-05],
...,
[3.1908e-04, 1.1903e-04, 8.0879e-05, ..., 3.8150e-06,
1.1389e-05, 1.6506e-05],
[3.3244e-04, 2.9054e-04, 2.2807e-04, ..., 5.9319e-06,
1.4453e-05, 8.5467e-05],
[3.3504e-04, 3.3504e-04, 3.3114e-04, ..., 1.5936e-05,
8.4637e-05, 1.2266e-04]]],
[[[3.3504e-04, 3.3244e-04, 3.1600e-04, ..., 1.1773e-04,
1.2242e-04, 3.0526e-04],
[3.3244e-04, 2.5697e-04, 1.0029e-04, ..., 3.8080e-05,
9.2776e-05, 1.2242e-04],
[3.1972e-04, 1.2336e-04, 1.2338e-04, ..., 1.0953e-05,
3.7931e-05, 9.3870e-05],
...,
[3.0474e-04, 3.9288e-05, 1.6701e-05, ..., 2.9772e-05,
8.4144e-05, 1.2194e-04],
[3.1686e-04, 9.2780e-05, 3.9442e-05, ..., 8.4144e-05,
1.2099e-04, 2.7904e-04],
[8.2772e-04, 2.7638e-04, 9.3507e-05, ..., 1.2194e-04,
2.7904e-04, 8.2608e-04]],
[[3.3374e-04, 2.7674e-04, 1.1771e-04, ..., 4.1335e-05,
1.0110e-04, 1.1773e-04],
[2.7894e-04, 1.0676e-04, 4.3824e-05, ..., 5.3381e-06,
2.9541e-05, 4.5044e-05],
[1.2338e-04, 1.1060e-04, 4.7945e-05, ..., 2.2603e-06,
1.2558e-05, 4.3318e-05],
...,
[4.3318e-05, 1.4510e-05, 6.1442e-06, ..., 6.1441e-06,
3.4001e-05, 9.3868e-05],
[4.5044e-05, 3.4135e-05, 1.4510e-05, ..., 3.4201e-05,
9.0804e-05, 1.2242e-04],
[2.7800e-04, 4.5044e-05, 4.3318e-05, ..., 9.3502e-05,
1.2242e-04, 7.5518e-04]],
[[1.2266e-04, 9.7795e-05, 3.4399e-05, ..., 4.0853e-05,
4.2981e-05, 1.1234e-04],
[1.1212e-04, 1.6701e-05, 1.6124e-05, ..., 4.3230e-06,
1.4119e-05, 4.3318e-05],
[1.1344e-04, 2.0462e-05, 1.3109e-05, ..., 1.7881e-06,
9.8567e-06, 3.4135e-05],
...,
[3.2827e-05, 5.5076e-06, 3.2632e-06, ..., 6.1441e-06,
1.4454e-05, 4.3318e-05],
[4.2981e-05, 1.1082e-05, 2.3412e-06, ..., 6.1441e-06,
3.0124e-05, 1.0189e-04],
[1.1235e-04, 3.7198e-05, 1.5207e-05, ..., 4.1335e-05,
1.0110e-04, 1.2290e-04]],
...,
[[1.2290e-04, 1.0269e-04, 4.5396e-05, ..., 4.2559e-06,
5.5508e-06, 3.7562e-05],
[1.1773e-04, 4.4519e-05, 3.9442e-05, ..., 2.6577e-07,
6.2279e-07, 5.5508e-06],
[9.3868e-05, 3.9288e-05, 1.6701e-05, ..., 1.1254e-07,
2.6577e-07, 4.2559e-06],
...,
[3.3533e-04, 2.9022e-04, 1.2339e-04, ..., 3.8821e-07,
1.7603e-06, 5.8628e-06],
[3.3533e-04, 3.2885e-04, 2.9136e-04, ..., 1.7062e-06,
4.5128e-06, 1.3792e-05],
[8.2935e-04, 3.3532e-04, 3.3533e-04, ..., 5.8628e-06,
6.0963e-06, 3.9441e-05]],
[[3.0526e-04, 1.2242e-04, 1.1773e-04, ..., 5.8628e-06,
1.3792e-05, 1.0229e-04],
[1.2338e-04, 1.0678e-04, 4.5397e-05, ..., 1.8960e-06,
4.6018e-06, 1.3792e-05],
[1.2194e-04, 8.7159e-05, 4.5397e-05, ..., 5.4533e-07,
1.8960e-06, 5.8628e-06],
...,
[8.6927e-04, 3.3533e-04, 3.3533e-04, ..., 2.2603e-06,
5.3381e-06, 4.1254e-05],
[8.6914e-04, 3.3533e-04, 3.3533e-04, ..., 4.3230e-06,
6.0018e-06, 4.1254e-05],
[9.0720e-04, 8.6910e-04, 6.9041e-04, ..., 6.0963e-06,
3.1384e-05, 1.1212e-04]],
[[3.3377e-04, 2.7904e-04, 1.2338e-04, ..., 4.1334e-05,
1.0110e-04, 1.2290e-04],
[2.7904e-04, 1.2338e-04, 1.2338e-04, ..., 5.9551e-06,
2.6688e-05, 1.0110e-04],
[1.2338e-04, 1.1912e-04, 9.7232e-05, ..., 4.0294e-06,
5.9551e-06, 4.1254e-05],
...,
[9.1076e-04, 8.7950e-04, 5.9762e-04, ..., 5.8172e-06,
2.7751e-05, 4.4868e-05],
[9.1075e-04, 9.1079e-04, 8.7941e-04, ..., 6.1441e-06,
3.4001e-05, 1.0269e-04],
[9.1075e-04, 9.1075e-04, 9.1075e-04, ..., 4.1335e-05,
1.0110e-04, 1.2290e-04]]]])
全部评论（2）
只看楼主
P
2024/7/15
没有更多内容了
--
楼主
0 点赞
0 帖子
楼主热帖
暂无数据常见问题新帖
暂无数据