| from .yowo.build import build_yowo | |
| def build_model(args, | |
| d_cfg, | |
| m_cfg, | |
| device, | |
| num_classes=3, | |
| trainable=False, | |
| resume=None): | |
| # build action detector | |
| if 'yowo_v2_' in args.version: | |
| model, criterion = build_yowo( | |
| args=args, | |
| d_cfg=d_cfg, | |
| m_cfg=m_cfg, | |
| device=device, | |
| num_classes=num_classes, | |
| trainable=trainable, | |
| resume=resume | |
| ) | |
| return model, criterion | |