import argparse
from modules.generate import interpolation

arg_lists = []
parser = argparse.ArgumentParser()

def add_argument_group(name):
    arg = parser.add_argument_group(name)
    arg_lists.append(arg)
    return arg


data_arg = add_argument_group('data')
data_arg.add_argument('--path', type=str, default='frames/')
data_arg.add_argument('--img_fmt', type=str, default='jpg')
data_arg.add_argument('--model', type=str, default='output.pth')
data_arg.add_argument('--run', type=int, default= 2) # example 1=2x 2=4x 3=8x ...
data_arg.add_argument('--batch_size', type=int, default= 4) # example 1=2x 2=4x 3=8x ...
data_arg.add_argument('--TensorRT', type=bool, default= False) # example 1=2x 2=4x 3=8x ...
data_arg.add_argument('--fp16', type=bool, default= False) # example 1=2x 2=4x 3=8x ...
startnum=0
args = parser.parse_args()

interpolation(batch_size=args.batch_size, temp_img = args.path, fp16=args.fp16, TensorRT=args.TensorRT, modelp=args.model,img_fmt=args.img_fmt)