code: def test_d4_image2mask_v2():
      x = torch.rand((4, 3, 224, 224))
      x_a = tta.d4_image_augment(x)
      y = tta.d4_image_deaugment(x_a)
  
      np.testing.assert_allclose(
          to_numpy(y), to_numpy(x), atol=1e-6, rtol=1e-6
      )
  
imports: from collections import defaultdict
  import cv2
  import numpy
  import pytest
  import torch
  from torch import nn
  from pytorch_toolbelt.inference import tta
  from pytorch_toolbelt.utils.torch_utils import to_numpy
label_desc: Use image node_id 4 for calling a global function with example usage: test_d4_image2mask_v2() without return types