| 123456789101112131415161718 |
- from .fcos import FcosE2E
- from .criterion import SetCriterion
- def build_fcos_e2e(cfg, is_val=False):
- # ------------ build object detector ------------
- ## RT-FCOS
- model = FcosE2E(cfg = cfg,
- conf_thresh = cfg.train_conf_thresh if is_val else cfg.test_conf_thresh,
- topk_results = cfg.train_topk if is_val else cfg.test_topk,
- )
- criterion = None
- if is_val:
- # build criterion for training
- criterion = SetCriterion(cfg)
- return model, criterion
-
|
