support gsam2 image predictor model
This commit is contained in:
18
grounding_dino/groundingdino/models/__init__.py
Normal file
18
grounding_dino/groundingdino/models/__init__.py
Normal file
@@ -0,0 +1,18 @@
|
||||
# ------------------------------------------------------------------------
|
||||
# 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
|
Reference in New Issue
Block a user