inference_sahi.py

from utils import *
def run_sahi_prediction(args, pt_model, source, imgsz ):
    predictor = compile_predictor(args, pt_modelpath=pt_model, save_dir = Path('./sahi/prediction/res_SAHI/'), sahi=True, imgsz = imgsz)
    LOGGER.info('Starting SAHI prediction')
    result = predictor(source = source)
    return result


def run_basic_prediction(pt_model, args, source):
    predictor = compile_predictor(args, pt_modelpath = pt_model, save_dir = Path('./sahi/prediction/res/'))
    LOGGER.info('Starting basic prediction')
    result = predictor(model = pt_model, source = source)
    return result