File size: 379 Bytes
c92c0ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import os
import sys

migc_path = os.path.dirname(os.path.abspath(__file__))
print(migc_path)
if migc_path not in sys.path:
    sys.path.append(migc_path)

from model_bbox.MIGC.inference_single_image import MIGC_Pipe

def load_local_model(model_name, model_type):
    if model_name == "MIGC":
        pipe = MIGC_Pipe()
    else:
        raise NotImplementedError
    return pipe