mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-22 13:40:47 +00:00
19 lines
754 B
Python
19 lines
754 B
Python
# ------------------------------------------------------------------------
|
|
# Grounding DINO
|
|
# url: https://github.com/IDEA-Research/GroundingDINO
|
|
# Copyright (c) 2023 IDEA. All Rights Reserved.
|
|
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
|
|
# ------------------------------------------------------------------------
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
|
from .GroundingDINO import build_groundingdino
|
|
|
|
|
|
def build_model(args):
|
|
# we use register to maintain models from catdet6 on.
|
|
from .registry import MODULE_BUILD_FUNCS
|
|
|
|
assert args.modelname in MODULE_BUILD_FUNCS._module_dict
|
|
build_func = MODULE_BUILD_FUNCS.get(args.modelname)
|
|
model = build_func(args)
|
|
return model
|