25 lines
982 B
Python
25 lines
982 B
Python
|
# ------------------------------------------------------------------------
|
||
|
|
# Deformable DETR
|
||
|
|
# Copyright (c) 2020 SenseTime. All Rights Reserved.
|
||
|
|
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
|
||
|
|
# ------------------------------------------------------------------------
|
||
|
|
# Modified from DETR (https://github.com/facebookresearch/detr)
|
||
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
||
|
|
# ------------------------------------------------------------------------
|
||
|
|
|
||
|
|
from .deformable_detr_single import build as build_single
|
||
|
|
from .deformable_detr_multi import build as build_multi
|
||
|
|
from .deformable_detr_multi_multi import build as build_multi_multi
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def build_model(args):
|
||
|
|
if args.dataset_file == "vid_single":
|
||
|
|
return build_single(args)
|
||
|
|
elif args.dataset_file in ["vid_multi", "vid_multi_mine"]:
|
||
|
|
return build_multi(args)
|
||
|
|
elif args.dataset_file == "vid_multi_mine_multi":
|
||
|
|
return build_multi_multi(args)
|
||
|
|
|
||
|
|
|