14 lines
317 B
Python
14 lines
317 B
Python
import argparse
|
|
|
|
from util import Args
|
|
from .classification import *
|
|
from .detection import *
|
|
|
|
|
|
def get_model(model_args):
|
|
model_args_ = model_args
|
|
|
|
if isinstance(model_args, argparse.Namespace):
|
|
model_args_ = Args(vars(model_args))
|
|
|
|
return globals().copy()[model_args_.get("model")](model_args_)
|