def input_fn()

in code/inference.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def input_fn(request_body, request_content_type):
    frame_width = 1024
    frame_height = 1024
    interval = 30
    f = io.BytesIO(request_body)
    tfile = tempfile.NamedTemporaryFile(delete=False)
    tfile.write(f.read())
    print(tfile.name)
    video_frames = video2frame(tfile,frame_width, frame_height, interval)  
    #convert to tensor of float32 type
    transform = transforms.Compose([
        transforms.Lambda(lambda video_frames: torch.stack([transforms.ToTensor()(frame) for frame in video_frames])) # returns        a 4D tensor
    ])
    image_tensors = transform(video_frames)

    return image_tensors