elazarcoh / simply-view-image-for-python-debugging

visual studio code extension simply view the image of the image variables when debugging python
https://marketplace.visualstudio.com/items?itemName=elazarcoh.simply-view-image-for-python-debugging
MIT License
57 stars 6 forks source link

Can not catch PyTorch tensors (and doesn't update variables) #46

Closed iNLyze closed 1 year ago

iNLyze commented 1 year ago

Is it normal that PyTorch Tensors are not shown as images? I sometimes get "Error running command svifpd.watch-refresh: Cannot read properties of undefined (reading 'isError'). This is likely caused by the extension that contributes svifpd.watch-refresh." when I try.

elazarcoh commented 1 year ago

Hmm.. maybe it's a regression in the new version.

Can you provide me with:

elazarcoh commented 1 year ago

Also, you can find the log under Output > View Image for Python, it can help. You might need to increase the verbosity in the settings of the extension.

iNLyze commented 1 year ago

Actually, I noticed, I can view the first four Tensor variables or so. After this any downstream Tensors will simply be ignored. The yellow light bulb won't show for them and I can't watch them in the Image Watch view.

Torch version: '1.13.1+cu117' Extension version: v3.0.0 Minimal example: In my example the images are provided through a dict. So I have a

img = batch["img"]
img_aug = batch["img_aug"]
coord_aug = batch["coord_aug"]
img_pos = batch["img_pos"]
label = batch["label"]
label_pos = batch["label_pos"]

I can view up until, say, img_pos. I am posting an example below. Typical images are: image

batch = {'ind': tensor([20547, 3695, 3241, 8605, 9354, 21599, 11628, 7024], device='cuda:0'), 'img': tensor([[[[ 0.6028, 1.3508, 2.0107, ..., -0.5410, -0.5850, -0.5410], [ 1.4827, 1.6147, 1.5707, ..., -0.5410, -0.4530, -0.5850], [ 1.0428, 1.8787, 0.8228, ..., -0.5850, -0.5850, -0.5410], ..., [-0.5850, -0.5850, -0.5850, ..., -0.5410, -0.4970, -0.5850], [-0.5850, -0.5850, -0.5850, ..., -0.5410, -0.5850, -0.4970], [-0.5410, -0.5850, -0.5850, ..., -0.4090, -0.5410, -0.4970]],

     [[-0.0445, -0.5766, -0.5766,  ..., -0.5471, -0.5471, -0.6062],
      [ 0.3399,  0.3695, -0.5766,  ..., -0.6062, -0.6062, -0.5175],
      [-0.5175, -0.4288, -0.5471,  ..., -0.6062, -0.6062, -0.3697],
      ...,
      [-0.6062, -0.6062, -0.6062,  ...,  0.4286, -0.4288,  0.4877],
      [-0.6062, -0.6062, -0.6062,  ..., -0.0149,  0.4582,  0.7243],
      [-0.6062, -0.6062, -0.6062,  ..., -0.3697, -0.1923, -0.1923]],

     [[ 0.0353, -0.0742, -0.2932,  ..., -0.3753, -0.3753, -0.3753],
      [-0.0468, -0.2658, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.1837, -0.3479, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ...,  1.1027,  0.7743,  0.5827],
      [-0.3753, -0.3753, -0.3753,  ...,  1.0754,  0.9111,  0.7195],
      [-0.3753, -0.3753, -0.3753,  ...,  0.9111,  0.9933,  0.6374]]],

    [[[ 0.4709,  0.1629,  1.6147,  ..., -0.1891,  0.2509,  0.9988],
      [ 2.0107,  2.0547,  0.6468,  ...,  1.5707,  2.6706,  3.1545],
      [ 0.5149, -0.0131,  0.9108,  ...,  0.3389, -0.1451, -0.1891],
      ...,
      [-0.4970, -0.4970, -0.4530,  ...,  0.3829, -0.4530, -0.4970],
      [-0.5410, -0.4970, -0.4530,  ...,  0.6028,  0.2949,  0.2069],
      [-0.4970, -0.5410, -0.4530,  ..., -0.4970, -0.3211, -0.3211]],

     [[-0.0149, -0.3697, -0.0445,  ..., -0.1331,  0.0147, -0.0149],
      [-0.1923, -0.0445, -0.1331,  ...,  0.0442,  0.3990, -0.4584],
      [-0.4288, -0.1923,  0.1625,  ..., -0.4288, -0.5175, -0.4288],
      ...,
      [-0.1627, -0.1627, -0.3105,  ...,  0.6060,  1.1086,  1.3451],
      [-0.0740, -0.0445,  0.1329,  ...,  0.6356,  3.3556,  1.0791],
      [ 0.1625,  0.0442,  0.1921,  ...,  2.4391,  2.1139,  1.3156]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3479, -0.3753, -0.3206],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3479, -0.3206],
      [-0.3479, -0.3479, -0.3753,  ..., -0.2932, -0.3753, -0.3206],
      ...,
      [ 0.1995,  0.1174, -0.0742,  ...,  2.0607,  1.3764,  2.4439],
      [ 0.1995,  0.1995, -0.0742,  ...,  0.7195,  1.5407,  2.8545],
      [ 1.0754,  0.9659,  0.3363,  ...,  0.3637,  1.1301,  2.4165]]],

    [[[-0.5850, -0.5850, -0.5410,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5410,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5410,  ..., -0.5850, -0.5850, -0.5850],
      ...,
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850]],

     [[-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.5766, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      ...,
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.5766],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    ...,

    [[[-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      ...,
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850]],

     [[-0.6062, -0.5766, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      ...,
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.5766, -0.6062,  ..., -0.6062, -0.5766, -0.3697],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.5175]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    [[[ 1.0868,  0.7348,  1.0868,  ...,  0.5149,  0.6468,  0.4709],
      [ 0.7788, -0.0131,  1.0868,  ..., -0.3211,  0.4269,  0.2509],
      [ 1.9227,  1.5267,  2.0547,  ...,  1.8787,  2.1867,  1.2628],
      ...,
      [ 1.3947,  1.8347,  3.0666,  ...,  3.1105,  2.5386,  0.2949],
      [ 4.4744,  3.7705,  2.2746,  ..., -0.2771, -0.2771,  0.2949],
      [ 2.7146,  2.0987,  4.6064,  ...,  2.4946, -0.4530, -0.5410]],

     [[-0.5766, -0.6062, -0.5175,  ..., -0.3105, -0.4584,  0.2808],
      [-0.5471, -0.5471, -0.5471,  ..., -0.4584, -0.6062, -0.3105],
      [-0.0740, -0.5471, -0.6062,  ..., -0.5471, -0.4288, -0.4879],
      ...,
      [-0.6062, -0.6062,  0.0147,  ..., -0.6062, -0.5471, -0.6062],
      [-0.0740, -0.5766, -0.5766,  ..., -0.5766, -0.5471, -0.6062],
      [-0.2514, -0.5471, -0.4879,  ..., -0.5471, -0.6062, -0.6062]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3206, -0.3479, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3479]]],

    [[[ 0.1189,  1.3508,  0.8228,  ...,  0.5588,  0.1629, -0.2331],
      [ 1.3068,  1.1748,  2.0107,  ...,  0.0309, -0.1011, -0.5410],
      [ 0.2509,  0.2069,  0.3389,  ...,  0.2069, -0.4970, -0.5410],
      ...,
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.4970, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850]],

     [[-0.3105, -0.3697, -0.3401,  ..., -0.4584, -0.5471, -0.4879],
      [-0.1627,  0.0147, -0.5175,  ..., -0.2218, -0.4584, -0.2218],
      [-0.3105, -0.2218, -0.5471,  ..., -0.4879, -0.4584, -0.6062],
      ...,
      [-0.6062, -0.4879, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.4879, -0.6062,  ..., -0.6062, -0.6062, -0.6062]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]]],
   device='cuda:0'), 'label': tensor([[[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    ...,

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]]], device='cuda:0'), 'img_pos': tensor([[[[-0.5850, -0.5850, -0.5850,  ..., -0.4970, -0.4530, -0.3211],
      [-0.5850, -0.5850, -0.5850,  ..., -0.4530, -0.1451, -0.2331],
      [-0.5410, -0.5850, -0.5850,  ...,  0.3389,  0.2069,  0.1189],
      ...,
      [-0.4530, -0.4530, -0.1891,  ..., -0.5850, -0.5850, -0.5410],
      [-0.4970, -0.5410, -0.2771,  ..., -0.5850, -0.5850, -0.5410],
      [-0.4970, -0.3650, -0.3211,  ..., -0.5410, -0.5850, -0.5850]],

     [[-0.5766, -0.6062, -0.5766,  ...,  0.2216, -0.2218, -0.3697],
      [-0.6062, -0.6062, -0.6062,  ...,  0.0442, -0.3697,  0.4582],
      [-0.6062, -0.5766, -0.6062,  ...,  0.1034, -0.3401, -0.1036],
      ...,
      [-0.6062, -0.6062, -0.4288,  ..., -0.6062, -0.6062, -0.6062],
      [-0.5175, -0.5175, -0.4879,  ..., -0.6062, -0.6062, -0.6062],
      [-0.5175, -0.5766, -0.4288,  ..., -0.5471, -0.6062, -0.6062]],

     [[-0.3753, -0.3753, -0.3753,  ...,  0.3090,  0.1174,  0.1721],
      [-0.3753, -0.3753, -0.3753,  ..., -0.0468, -0.1563, -0.1290],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3206, -0.2932],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    [[[ 0.9548,  1.1748, -0.4970,  ...,  3.2865,  5.1343,  6.3662],
      [ 1.6147,  1.0868, -0.4970,  ..., 10.4137,  6.0582,  2.8026],
      [ 0.3389,  1.9667, -0.0131,  ...,  3.8145,  6.6301,  8.2579],
      ...,
      [ 0.0309,  0.4269,  0.6028,  ..., -0.5850, -0.5850, -0.5850],
      [ 0.6028,  1.0428,  0.7348,  ..., -0.5850, -0.5850, -0.5850],
      [ 0.9988,  1.2628,  1.3068,  ..., -0.5850, -0.5850, -0.5850]],

     [[ 1.1086,  0.4286,  0.9608,  ...,  0.4582,  0.0738, -0.1923],
      [-0.1923,  0.1329,  0.4286,  ...,  1.1382, -0.1923, -0.2810],
      [-0.2218,  1.1678,  1.6999,  ...,  0.3103,  0.3399,  2.3504],
      ...,
      [-0.1923, -0.1923, -0.3105,  ..., -0.6062, -0.6062, -0.6062],
      [-0.3105, -0.3697, -0.3401,  ..., -0.6062, -0.6062, -0.6062],
      [ 0.3695, -0.0740, -0.3401,  ..., -0.6062, -0.6062, -0.6062]],

     [[-0.3479, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3479, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3479, -0.3479,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3479,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    [[[-0.5410, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5410, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5410],
      [-0.5850, -0.5410, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      ...,
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.5850],
      [-0.5410, -0.5850, -0.5850,  ..., -0.5410, -0.5850, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5410, -0.5850, -0.5850]],

     [[-0.6062, -0.6062, -0.5766,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      ...,
      [-0.6062, -0.6062, -0.6062,  ..., -0.5766, -0.5766, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.6062, -0.6062]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    ...,

    [[[-0.5850, -0.5850, -0.5850,  ..., -0.1451,  0.0309,  0.1629],
      [-0.5410, -0.5850, -0.5850,  ..., -0.4090, -0.4530, -0.1891],
      [-0.5410, -0.5850, -0.5850,  ..., -0.5410, -0.4090, -0.2331],
      ...,
      [-0.4970, -0.4530, -0.3211,  ..., -0.4530, -0.4090, -0.0131],
      [-0.4090, -0.4530, -0.5410,  ..., -0.4970, -0.5410, -0.3211],
      [-0.4970, -0.4970, -0.4970,  ..., -0.4530, -0.3650, -0.5410]],

     [[-0.5175, -0.5766, -0.6062,  ..., -0.5766, -0.5766, -0.5766],
      [-0.3992, -0.5766, -0.6062,  ..., -0.4288, -0.6062, -0.5175],
      [-0.3697, -0.6062, -0.5766,  ..., -0.6062, -0.6062, -0.5766],
      ...,
      [-0.6062, -0.6062, -0.6062,  ...,  2.2913,  0.5764,  3.1782],
      [-0.6062, -0.6062, -0.5471,  ...,  2.2913,  3.0895,  1.4338],
      [-0.6062, -0.6062, -0.6062,  ...,  5.4548,  1.4930,  0.8130]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ...,  1.1575,  1.4038,  1.5407],
      [-0.3753, -0.3753, -0.3753,  ...,  1.1848,  0.7469,  0.9933],
      [-0.3753, -0.3753, -0.3753,  ...,  0.4458,  0.8017,  1.1301]]],

    [[[-0.5410, -0.5850, -0.5850,  ...,  0.4269,  0.4709,  0.4269],
      [-0.5850, -0.5850, -0.5850,  ...,  0.2949,  0.0309, -0.0571],
      [-0.4970, -0.5850, -0.5850,  ...,  0.7788,  0.3389, -0.1011],
      ...,
      [-0.5850, -0.5850, -0.5850,  ...,  0.8228,  1.8347,  1.3068],
      [-0.5850, -0.5850, -0.5850,  ...,  1.6147,  1.2628,  0.1189],
      [-0.5850, -0.5850, -0.5850,  ...,  0.6028,  0.1189, -0.1011]],

     [[-0.6062, -0.6062, -0.6062,  ..., -0.5471, -0.5766, -0.5175],
      [-0.6062, -0.6062, -0.6062,  ..., -0.6062, -0.5471, -0.5766],
      [-0.6062, -0.5766, -0.6062,  ..., -0.4584, -0.1923, -0.5766],
      ...,
      [-0.6062, -0.6062, -0.6062,  ..., -0.5471, -0.3697, -0.5471],
      [-0.6062, -0.6062, -0.6062,  ..., -0.5175, -0.2810, -0.5175],
      [-0.6062, -0.6062, -0.6062,  ..., -0.5766, -0.5471, -0.5766]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3479, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3479],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3479, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]],

    [[[-0.1011, -0.1011, -0.1891,  ..., -0.5850, -0.5850, -0.5850],
      [-0.2771, -0.2771, -0.2331,  ..., -0.5850, -0.5850, -0.5850],
      [-0.2771, -0.1451, -0.3650,  ..., -0.5850, -0.5850, -0.5850],
      ...,
      [-0.5410, -0.5410, -0.5850,  ..., -0.5850, -0.4970, -0.5850],
      [-0.5850, -0.5410, -0.5410,  ..., -0.5850, -0.5410, -0.5850],
      [-0.5850, -0.5850, -0.5850,  ..., -0.5850, -0.5850, -0.4970]],

     [[-0.5175, -0.6062, -0.6062,  ..., -0.6062, -0.5471, -0.6062],
      [-0.6062, -0.6062, -0.5766,  ..., -0.5471, -0.6062, -0.6062],
      [-0.6062, -0.5175, -0.6062,  ..., -0.6062, -0.6062, -0.6062],
      ...,
      [-0.6062, -0.6062, -0.5471,  ..., -0.6062, -0.5766, -0.6062],
      [-0.6062, -0.6062, -0.5471,  ..., -0.6062, -0.6062, -0.6062],
      [-0.5766, -0.6062, -0.6062,  ..., -0.5766, -0.6062, -0.5471]],

     [[-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      ...,
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753],
      [-0.3753, -0.3753, -0.3753,  ..., -0.3753, -0.3753, -0.3753]]]],
   device='cuda:0'), 'ind_pos': tensor([ 7109, 12807, 12253,  8626,   590, 19794, 12249,  3210],
   device='cuda:0'), 'mask': tensor([[[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    ...,

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]]], device='cuda:0'), 'label_pos': tensor([[[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    ...,

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]],

    [[-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     ...,
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1],
     [-1, -1, -1,  ..., -1, -1, -1]]], device='cuda:0'), 'mask_pos': tensor([[[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    ...,

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]],

    [[0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     ...,
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.],
     [0., 0., 0.,  ..., 0., 0., 0.]]], device='cuda:0'), 'img_aug': tensor([[[[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [8.2373e-01, 8.2373e-01, 8.2373e-01,  ..., 8.0667e-01,
       8.2351e-01, 8.2373e-01],
      [8.1628e-01, 8.2352e-01, 8.2349e-01,  ..., 8.1815e-01,
       8.2373e-01, 8.2373e-01],
      [5.5447e-01, 8.1628e-01, 8.1521e-01,  ..., 8.1444e-01,
       8.2373e-01, 8.2373e-01]],

     [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [5.5208e-02, 1.5271e-03, 3.2320e-08,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [1.5271e-03, 4.2241e-05, 8.9401e-10,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [6.4641e-08, 1.7880e-09, 3.7842e-14,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [1.0270e-02, 1.0948e-01, 1.1286e-01,  ..., 1.0780e-01,
       1.1221e-01, 1.1226e-01],
      [1.1146e-01, 1.1885e-01, 1.4337e-01,  ..., 1.0915e-01,
       1.1221e-01, 1.1124e-01],
      [1.8317e-01, 3.3791e-01, 4.9140e-01,  ..., 1.0123e-01,
       1.1203e-01, 1.1223e-01]]],

    [[[9.2444e-01, 8.1992e-01, 3.7691e-01,  ..., 8.4269e-01,
       2.3908e-01, 0.0000e+00],
      [9.7069e-01, 9.7069e-01, 9.0643e-01,  ..., 4.4023e-01,
       0.0000e+00, 4.1725e-02],
      [8.1992e-01, 8.5645e-01, 8.4821e-01,  ..., 8.5284e-01,
       8.4671e-01, 8.8362e-01],
      ...,
      [0.0000e+00, 8.2598e-02, 8.2041e-01,  ..., 1.2516e-01,
       1.7915e-01, 0.0000e+00],
      [0.0000e+00, 1.5602e-02, 8.4044e-01,  ..., 2.4848e-02,
       0.0000e+00, 0.0000e+00],
      [8.1992e-01, 8.2724e-01, 8.1992e-01,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[6.9643e-01, 7.6772e-02, 3.5292e-02,  ..., 2.1174e-01,
       2.2385e-02, 0.0000e+00],
      [9.7069e-01, 9.7069e-01, 5.8969e-01,  ..., 4.6310e-02,
       0.0000e+00, 3.9068e-03],
      [7.6772e-02, 2.9335e-01, 2.4448e-01,  ..., 2.7192e-01,
       2.3557e-01, 4.5446e-01],
      ...,
      [0.0000e+00, 7.7339e-03, 7.9650e-02,  ..., 1.1719e-02,
       1.6775e-02, 0.0000e+00],
      [0.0000e+00, 1.4609e-03, 1.9844e-01,  ..., 2.3266e-03,
       0.0000e+00, 0.0000e+00],
      [7.6772e-02, 1.2013e-01, 7.6772e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[1.8128e-01, 7.6772e-02, 3.5292e-02,  ..., 9.9536e-02,
       2.2385e-02, 0.0000e+00],
      [2.2754e-01, 2.2754e-01, 1.6328e-01,  ..., 4.2011e-02,
       0.0000e+00, 3.9068e-03],
      [7.6772e-02, 1.1330e-01, 1.0506e-01,  ..., 1.0969e-01,
       1.0356e-01, 1.4047e-01],
      ...,
      [0.0000e+00, 7.7339e-03, 7.7257e-02,  ..., 1.1719e-02,
       1.6775e-02, 0.0000e+00],
      [0.0000e+00, 1.4609e-03, 9.7292e-02,  ..., 2.3266e-03,
       0.0000e+00, 0.0000e+00],
      [7.6772e-02, 8.4084e-02, 7.6772e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]]],

    [[[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]]],

    ...,

    [[[6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      ...,
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7711e-01],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 1.8511e-01],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 4.5691e-01]],

     [[6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      ...,
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 1.9219e-01],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 9.1577e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 1.4716e-01]],

     [[6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 6.7534e-02],
      ...,
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 2.3314e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 5.9005e-02],
      [6.7534e-02, 6.7534e-02, 6.7534e-02,  ..., 6.7534e-02,
       6.7534e-02, 3.9288e-02]]],

    [[[6.8439e-01, 7.0293e-01, 7.7691e-01,  ..., 8.2570e-01,
       8.5328e-01, 8.7242e-01],
      [6.5747e-01, 6.7879e-01, 7.5152e-01,  ..., 7.3760e-01,
       7.5843e-01, 7.7314e-01],
      [6.0974e-01, 6.2333e-01, 6.6523e-01,  ..., 5.7977e-01,
       5.9060e-01, 5.9851e-01],
      ...,
      [1.2894e-01, 1.0606e-01, 9.2130e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [1.1886e-01, 9.7957e-02, 8.4272e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [1.1469e-01, 9.5865e-02, 7.7139e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[1.8073e-01, 1.6340e-01, 1.5348e-01,  ..., 2.0946e-02,
       2.1645e-02, 2.2131e-02],
      [2.6743e-01, 2.5670e-01, 2.4782e-01,  ..., 1.8711e-02,
       1.9239e-02, 1.9612e-02],
      [4.3542e-01, 4.3082e-01, 4.3666e-01,  ..., 1.4707e-02,
       1.4982e-02, 1.5182e-02],
      ...,
      [3.2709e-03, 2.6904e-03, 2.3371e-03,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [3.0153e-03, 2.4849e-03, 2.1378e-03,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [2.9094e-03, 2.4318e-03, 1.9568e-03,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[9.4750e-01, 9.3632e-01, 9.0972e-01,  ..., 1.2965e-01,
       1.3398e-01, 1.3699e-01],
      [9.4416e-01, 9.3528e-01, 9.1337e-01,  ..., 1.1582e-01,
       1.1909e-01, 1.2140e-01],
      [9.4328e-01, 9.3665e-01, 9.1914e-01,  ..., 9.1036e-02,
       9.2737e-02, 9.3979e-02],
      ...,
      [2.0247e-02, 1.6654e-02, 1.4466e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [1.8664e-02, 1.5381e-02, 1.3233e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [1.8009e-02, 1.5053e-02, 1.2112e-02,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]]],

    [[[0.0000e+00, 1.9020e-01, 0.0000e+00,  ..., 6.9544e-01,
       9.2635e-01, 9.2635e-01],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 8.9397e-01,
       9.2635e-01, 9.2635e-01],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 4.6602e-01,
       9.2635e-01, 3.6675e-01],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[0.0000e+00, 2.1197e-02, 0.0000e+00,  ..., 7.7506e-02,
       1.0324e-01, 1.0324e-01],
      [0.0000e+00, 0.0000e+00, 5.0501e-02,  ..., 9.9632e-02,
       1.0324e-01, 1.0324e-01],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 5.1938e-02,
       1.0324e-01, 4.0874e-02],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]],

     [[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 5.6283e-03,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      ...,
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00],
      [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00,
       0.0000e+00, 0.0000e+00]]]], device='cuda:0'), 'coord_aug': tensor([[[[-0.8708,  0.9726],
      [-0.8708,  0.9691],
      [-0.8708,  0.9655],
      ...,
      [-0.8708, -0.8794],
      [-0.8708, -0.8830],
      [-0.8708, -0.8865]],

     [[-0.8676,  0.9726],
      [-0.8676,  0.9691],
      [-0.8676,  0.9655],
      ...,
      [-0.8676, -0.8794],
      [-0.8676, -0.8830],
      [-0.8676, -0.8865]],

     [[-0.8642,  0.9726],
      [-0.8642,  0.9691],
      [-0.8642,  0.9655],
      ...,
      [-0.8642, -0.8794],
      [-0.8642, -0.8830],
      [-0.8642, -0.8865]],

     ...,

     [[ 0.8799,  0.9726],
      [ 0.8799,  0.9691],
      [ 0.8799,  0.9655],
      ...,
      [ 0.8799, -0.8794],
      [ 0.8799, -0.8830],
      [ 0.8799, -0.8865]],

     [[ 0.8833,  0.9726],
      [ 0.8833,  0.9691],
      [ 0.8833,  0.9655],
      ...,
      [ 0.8833, -0.8794],
      [ 0.8833, -0.8830],
      [ 0.8833, -0.8865]],

     [[ 0.8865,  0.9726],
      [ 0.8865,  0.9691],
      [ 0.8865,  0.9655],
      ...,
      [ 0.8865, -0.8794],
      [ 0.8865, -0.8830],
      [ 0.8865, -0.8865]]],

    [[[-0.9765, -0.8004],
      [-0.9765, -0.7971],
      [-0.9765, -0.7936],
      ...,
      [-0.9765,  0.9854],
      [-0.9765,  0.9889],
      [-0.9765,  0.9922]],

     [[-0.9729, -0.8004],
      [-0.9729, -0.7971],
      [-0.9729, -0.7936],
      ...,
      [-0.9729,  0.9854],
      [-0.9729,  0.9889],
      [-0.9729,  0.9922]],

     [[-0.9693, -0.8004],
      [-0.9693, -0.7971],
      [-0.9693, -0.7936],
      ...,
      [-0.9693,  0.9854],
      [-0.9693,  0.9889],
      [-0.9693,  0.9922]],

     ...,

     [[ 0.8988, -0.8004],
      [ 0.8988, -0.7971],
      [ 0.8988, -0.7936],
      ...,
      [ 0.8988,  0.9854],
      [ 0.8988,  0.9889],
      [ 0.8988,  0.9922]],

     [[ 0.9025, -0.8004],
      [ 0.9025, -0.7971],
      [ 0.9025, -0.7936],
      ...,
      [ 0.9025,  0.9854],
      [ 0.9025,  0.9889],
      [ 0.9025,  0.9922]],

     [[ 0.9061, -0.8004],
      [ 0.9061, -0.7971],
      [ 0.9061, -0.7936],
      ...,
      [ 0.9061,  0.9854],
      [ 0.9061,  0.9889],
      [ 0.9061,  0.9922]]],

    [[[-0.9374, -0.9139],
      [-0.9374, -0.9103],
      [-0.9374, -0.9066],
      ...,
      [-0.9374,  0.9770],
      [-0.9374,  0.9807],
      [-0.9374,  0.9843]],

     [[-0.9338, -0.9139],
      [-0.9338, -0.9103],
      [-0.9338, -0.9066],
      ...,
      [-0.9338,  0.9770],
      [-0.9338,  0.9807],
      [-0.9338,  0.9843]],

     [[-0.9300, -0.9139],
      [-0.9300, -0.9103],
      [-0.9300, -0.9066],
      ...,
      [-0.9300,  0.9770],
      [-0.9300,  0.9807],
      [-0.9300,  0.9843]],

     ...,

     [[ 0.9574, -0.9139],
      [ 0.9574, -0.9103],
      [ 0.9574, -0.9066],
      ...,
      [ 0.9574,  0.9770],
      [ 0.9574,  0.9807],
      [ 0.9574,  0.9843]],

     [[ 0.9611, -0.9139],
      [ 0.9611, -0.9103],
      [ 0.9611, -0.9066],
      ...,
      [ 0.9611,  0.9770],
      [ 0.9611,  0.9807],
      [ 0.9611,  0.9843]],

     [[ 0.9648, -0.9139],
      [ 0.9648, -0.9103],
      [ 0.9648, -0.9066],
      ...,
      [ 0.9648,  0.9770],
      [ 0.9648,  0.9807],
      [ 0.9648,  0.9843]]],

    ...,

    [[[-1.0000,  0.9883],
      [-1.0000,  0.9847],
      [-1.0000,  0.9809],
      ...,
      [-1.0000, -0.8988],
      [-1.0000, -0.9025],
      [-1.0000, -0.9061]],

     [[-0.9968,  0.9883],
      [-0.9968,  0.9847],
      [-0.9968,  0.9809],
      ...,
      [-0.9968, -0.8988],
      [-0.9968, -0.9025],
      [-0.9968, -0.9061]],

     [[-0.9933,  0.9883],
      [-0.9933,  0.9847],
      [-0.9933,  0.9809],
      ...,
      [-0.9933, -0.8988],
      [-0.9933, -0.9025],
      [-0.9933, -0.9061]],

     ...,

     [[ 0.7663,  0.9883],
      [ 0.7663,  0.9847],
      [ 0.7663,  0.9809],
      ...,
      [ 0.7663, -0.8988],
      [ 0.7663, -0.9025],
      [ 0.7663, -0.9061]],

     [[ 0.7697,  0.9883],
      [ 0.7697,  0.9847],
      [ 0.7697,  0.9809],
      ...,
      [ 0.7697, -0.8988],
      [ 0.7697, -0.9025],
      [ 0.7697, -0.9061]],

     [[ 0.7730,  0.9883],
      [ 0.7730,  0.9847],
      [ 0.7730,  0.9809],
      ...,
      [ 0.7730, -0.8988],
      [ 0.7730, -0.9025],
      [ 0.7730, -0.9061]]],

    [[[-0.9530, -0.8982],
      [-0.9530, -0.8946],
      [-0.9530, -0.8909],
      ...,
      [-0.9530,  0.9888],
      [-0.9530,  0.9925],
      [-0.9530,  0.9961]],

     [[-0.9494, -0.8982],
      [-0.9494, -0.8946],
      [-0.9494, -0.8909],
      ...,
      [-0.9494,  0.9888],
      [-0.9494,  0.9925],
      [-0.9494,  0.9961]],

     [[-0.9457, -0.8982],
      [-0.9457, -0.8946],
      [-0.9457, -0.8909],
      ...,
      [-0.9457,  0.9888],
      [-0.9457,  0.9925],
      [-0.9457,  0.9961]],

     ...,

     [[ 0.9340, -0.8982],
      [ 0.9340, -0.8946],
      [ 0.9340, -0.8909],
      ...,
      [ 0.9340,  0.9888],
      [ 0.9340,  0.9925],
      [ 0.9340,  0.9961]],

     [[ 0.9377, -0.8982],
      [ 0.9377, -0.8946],
      [ 0.9377, -0.8909],
      ...,
      [ 0.9377,  0.9888],
      [ 0.9377,  0.9925],
      [ 0.9377,  0.9961]],

     [[ 0.9413, -0.8982],
      [ 0.9413, -0.8946],
      [ 0.9413, -0.8909],
      ...,
      [ 0.9413,  0.9888],
      [ 0.9413,  0.9925],
      [ 0.9413,  0.9961]]],

    [[[-0.9726,  1.0000],
      [-0.9726,  0.9961],
      [-0.9726,  0.9923],
      ...,
      [-0.9726, -0.9727],
      [-0.9726, -0.9766],
      [-0.9726, -0.9804]],

     [[-0.9689,  1.0000],
      [-0.9689,  0.9961],
      [-0.9689,  0.9923],
      ...,
      [-0.9689, -0.9727],
      [-0.9689, -0.9766],
      [-0.9689, -0.9804]],

     [[-0.9651,  1.0000],
      [-0.9651,  0.9961],
      [-0.9651,  0.9923],
      ...,
      [-0.9651, -0.9727],
      [-0.9651, -0.9766],
      [-0.9651, -0.9804]],

     ...,

     [[ 0.9573,  1.0000],
      [ 0.9573,  0.9961],
      [ 0.9573,  0.9923],
      ...,
      [ 0.9573, -0.9727],
      [ 0.9573, -0.9766],
      [ 0.9573, -0.9804]],

     [[ 0.9610,  1.0000],
      [ 0.9610,  0.9961],
      [ 0.9610,  0.9923],
      ...,
      [ 0.9610, -0.9727],
      [ 0.9610, -0.9766],
      [ 0.9610, -0.9804]],

     [[ 0.9648,  1.0000],
      [ 0.9648,  0.9961],
      [ 0.9648,  0.9923],
      ...,
      [ 0.9648, -0.9727],
      [ 0.9648, -0.9766],
      [ 0.9648, -0.9804]]]], device='cuda:0')}
elazarcoh commented 1 year ago

Thanks. This is helpful. Hopefully, I'll find the reason/solution soon.

iNLyze commented 1 year ago

Here is the output just after the issue occurred: view_image_output.zip

iNLyze commented 1 year ago

Hey, and great idea to this extension! I really appreciate your effort. This extension could be a big time saver.

elazarcoh commented 1 year ago

I think the issue was a cause of something else. Try the new release, 3.0.1 If there are still bugs, please let me know.

cattaneod commented 1 year ago

Sorry to open a closed issue, but I'd like to inform that the new update (3.0.1) broke the extension for me. Now variables and expressions are not shown in the image watch menu. If i try to "Refresh", I get this error:

Error running command svifpd.watch-refresh: Cannot read properties of undefined (reading 'isError'). This is likely caused by the extension that contributes svifpd.watch-refresh.

Reverting to 3.0.0 solve the issue.

elazarcoh commented 1 year ago

@cattaneod Thanks for reporting. Can you please open a new issue, with a minimal example if possible?

elazarcoh commented 1 year ago

@cattaneod (hopefully) fixed in version 3.0.2. Please let me know if there are more issues.

MYTHOFLEGEND commented 1 year ago

I also met this problem with version 3.0.2. Most of the times, the variables cannot show on IMAGE WATCH, sometimes they can show but with Not Ready status. And when I refresh it, I got this message:

运行命令 svifpd.watch-refresh 错误: Cannot read properties of undefined (reading 'isError')。这可能是由提交 svifpd.watch-refresh 的扩展引起的。

MYTHOFLEGEND commented 1 year ago

I am debugging on a server by ssh, maybe that's the reason I guess

elazarcoh commented 1 year ago

ssh shouldn't be a problem. Does it happens in a minimal example I can try myself? can you provide a log (there's a log in the Output tab, you might need to enable it in the settings).

MYTHOFLEGEND commented 1 year ago

I tried to change "Svifpd: Debug" this setting, but I can only get info like this:

运行命令 svifpd.watch-refresh 错误: Cannot read properties of undefined (reading 'isError')。这可能是由提交 svifpd.watch-refresh 的扩展引起的。

image

my vscode version is 1.76.2 win x64, and the server is centos7

elazarcoh commented 1 year ago

after setting it to verbose, look for it here: image

MYTHOFLEGEND commented 1 year ago

logs.txt

MYTHOFLEGEND commented 1 year ago

maybe it is because tensorflow, when I try a normal script, it works fine

elazarcoh commented 1 year ago

Can you try install matplotlib? I know it should not be necessary to have, but there might be some bug in the code that assume it.

MYTHOFLEGEND commented 1 year ago

after I installed matplotlib, it works well 😂

elazarcoh commented 1 year ago

Great! Thanks for helping finding the issue!