10 lines
325 B
Python
10 lines
325 B
Python
|
from .bua import add_bottom_up_attention_config
|
||
|
|
|
||
|
|
def add_config(args, cfg):
|
||
|
|
if args.mode == "caffe":
|
||
|
|
add_bottom_up_attention_config(cfg, True)
|
||
|
|
elif args.mode == "detectron2":
|
||
|
|
add_bottom_up_attention_config(cfg)
|
||
|
|
else:
|
||
|
|
raise Exception("detection model not supported: {}".format(args.model))
|