Multipart form format about yolov5_torchserve HOT 3 CLOSED

def get_predictions(raw_frames_path: List[Path],
                    batch_size: int
                    ):
    batch = {}
    cpt = 1
    predictions = []
    images_paths = [p.as_posix() for p in raw_frames_path]
    logging.warning(f"images_paths {str(images_paths)}")
    for frame in sorted(images_paths):
        key = "img" + str(cpt)
        cpt += 1
        batch[key] = open(frame, 'rb')
        if len(batch.keys()) >= batch_size:
            # call torchserve predictions
            predictions += _get_prediction(batch)
            batch = {}  # reset batch
            cpt = 1  # reset image idx
    if len(batch.keys()) > 0:
        predictions += _get_prediction(batch)
    return predictions