def load_img(args):
    img_path = args
    img = cv2.imread(img_path)
    img = cv2.resize(img, (input_size, input_size))
    img = randomHueSaturationValue(img, hue_shift_limit=((- 5), 5), sat_shift_limit=((- 1), 1), val_shift_limit=((- 2), 2), u=0.5)
    img = randomShiftScaleRotate(img, shift_limit=((- 0.2), 0.2), scale_limit=((- 0.2), 0.5), rotate_limit=((- 30), 30), aspect_limit=((- 0.2), 0.2), u=0.5)
    img = randomHorizontalFlip(img)
    img = randomVerticalFlip(img)
    return img
def load_img(args):
    img_path = args
    img = cv2.imread(img_path)
    img = cv2.resize(img, (input_size, input_size))
    img = randomHueSaturationValue(img, hue_shift_limit=((- 5), 5), sat_shift_limit=((- 1), 1), val_shift_limit=((- 2), 2), u=0.5)
    img = randomShiftScaleRotate(img, shift_limit=((- 0.2), 0.2), scale_limit=((- 0.2), 0.5), rotate_limit=((- 30), 30), aspect_limit=((- 0.2), 0.2), u=0.5)
    img = randomHorizontalFlip(img)
    img = randomVerticalFlip(img)
    return img