code: def __init__(
      self, p: float = 0.3, upper_body_ids=None, lower_body_ids=None
  ):
      assert upper_body_ids is not None
      assert lower_body_ids is not None
      self.p = p
      self.upper_body_ids = upper_body_ids
      self.lower_body_ids = lower_body_ids
  
imports: import math
  import random
  from typing import Tuple
  import cv2
  import numpy
  import torch
  from torchvision.transforms import functional
  import matplotlib.pyplot
label_desc: Use image node_id 1 to create a new HalfBody object from inherited base classes: object with example: obj = HalfBody(p, upper_body_ids, lower_body_ids)