You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
754 B
19 lines
754 B
2 years ago
|
# ------------------------------------------------------------------------
|
||
|
|
# 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
|