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.
18 lines
754 B
18 lines
754 B
# ------------------------------------------------------------------------ |
|
# 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
|
|
|