YOLO conversion tool through API call

Hi @ThomasVoets,

Here's a demo how you can call the tools using an API call:

from typing import Union, Tuple, Literal, Dict
import requests
from uuid import uuid4
import os


STATUS_OK: int = 200
OUTPUT_FILE_NAME: str = "output.zip"
DEFAULT_NSHAVES: int = 6
DEFAULT_USE_LEGACY_FRONTEND: str = 'true'
DEFAULT_USE_RVC2: str = 'true'
URL_V7: str = "/yolov7"
URL_V6R3: str = "/yolov6r3"
URL_V6R1: str = "/yolov6r1"
DEFAULT_URL: str = "https://tools.luxonis.com"


def convert_yolo(file_path: str, shape: Union[int, Tuple[int, int]]=416, version: Literal["v5", "v6", "v6r2", "v6r4", "v7", "v8", "v9", "v10", "v11", "goldyolo"] = "v5", 
                    url: str=DEFAULT_URL, file_name:str=OUTPUT_FILE_NAME, log: bool=False, n_shaves: int=DEFAULT_NSHAVES, use_legacy: str=DEFAULT_USE_LEGACY_FRONTEND, use_rvc2: str=DEFAULT_USE_RVC2):
    """ Uploads Yolo weights and receives zip with compiled blob.
    
    :param file_path: Path to .pt weights
    :param shape: Integer or tuple with width and height - must be divisible by 32
    :param version: Version of the Yolo model
    :param url: tools' URL
    :param file_name: Output file
    :param log: Whether to switch on logging or not
    :param n_shaves: Number of shaves
    :param use_legacy: Whether to use the legacy frontend flag while compiling to IR representation or not
    :param use_rvc2: Whether to use the RVC2 or RVC3 conversion
    :returns: Status code and Path to downloaded zip file
    """
    # Variables
    with open(file_path,'rb') as input_file:
        files = {'file': input_file}
        values = {
            'inputshape': shape if isinstance(shape, int) else " ".join(map(str,shape)), 
            'version': version, 
            'id': uuid4(),
            'nShaves': n_shaves,
            'useLegacyFrontend': use_legacy,
            "useRVC2": use_rvc2
        }
        url = f"{url}/upload"
        if log:
            print(url)
        
        # Upload files
        session = requests.Session()
        try: 
            with session.post(url, files=files, data=values, stream=True) as r:
                r.raise_for_status()
                if log:
                    print(f"Conversion complete. Downloading...")
                
                with open(file_name, 'wb') as f:
                    for chunk in r.iter_content(chunk_size=8192): 
                        # If you have chunk encoded response uncomment if
                        # and set chunk_size parameter to None.
                        #if chunk: 
                        f.write(chunk)
                if log:
                    print(r.status_code)
            session.close()
            return r.status_code, file_name
        except Exception as e:
            print(e)
            session.close()
            return 500, ""


if __name__ == "__main__":
    convert_yolo(file_path="yolov6nr2.pt", shape=416, version="v6r2", url=f"{DEFAULT_URL}{URL_V6R3}")  

Best,
Jan