Args: cfg (CfgNode): a detectron2 config Returns: CfgNode: an updated config with new options that will be used by :class:`Caffe2Tracer`.
(cfg)
| 23 | |
| 24 | |
| 25 | def add_export_config(cfg): |
| 26 | """ |
| 27 | Args: |
| 28 | cfg (CfgNode): a detectron2 config |
| 29 | |
| 30 | Returns: |
| 31 | CfgNode: an updated config with new options that will be used |
| 32 | by :class:`Caffe2Tracer`. |
| 33 | """ |
| 34 | is_frozen = cfg.is_frozen() |
| 35 | cfg.defrost() |
| 36 | cfg.EXPORT_CAFFE2 = CN() |
| 37 | cfg.EXPORT_CAFFE2.USE_HEATMAP_MAX_KEYPOINT = False |
| 38 | if is_frozen: |
| 39 | cfg.freeze() |
| 40 | return cfg |
| 41 | |
| 42 | |
| 43 | class Caffe2Tracer: |