allow tracing pointrend models
@@ -13,6 +13,7 @@ from detectron2.evaluation import COCOEvaluator, inference_on_dataset, print_csv
from detectron2.export import Caffe2Tracer, add_export_config
from detectron2.export.torchscript import dump_torchscript_IR, export_torchscript_with_instances
from detectron2.modeling import GeneralizedRCNN, build_model
+from detectron2.projects.point_rend import add_pointrend_config
from detectron2.structures import Boxes
from detectron2.utils.env import TORCH_VERSION
from detectron2.utils.logger import setup_logger
@@ -23,6 +24,7 @@ def setup_cfg(args):
# cuda context is initialized before creating dataloader, so we don't fork anymore
cfg.DATALOADER.NUM_WORKERS = 0
cfg = add_export_config(cfg)
+ add_pointrend_config(cfg)
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
cfg.freeze()
