code: def l2_distance(a, b):
      dist = np.sqrt(np.sum(np.square(a - b), axis=1))
      distance_list = np.sqrt(((a - b) ** 2).sum(axis=2)).mean(axis=1)
      return distance_list
  
imports: from dataset import FaceDataset, DataLoaderX, MXFaceDataset
  import argparse
  import logging
  import os
  import time
  import timm
  import glob
  import numpy
  import os.path
  from utils.utils_config import get_config
  from scipy.spatial.transform import Rotation
  import torch
  import torch.distributed
  from torch import nn
  import torch.nn.functional
  import torch.utils.data.distributed
  from backbones import get_network
  from inference_simple import JMLRInference, Rt_from_6dof
  from dataset import Rt26dof
label_desc: Use image node_id 1 for calling a global function with example usage: l2_distance(a, b) and returns: distance_list