[Enhancement] Fix the build_segmentor in colab demo (#1551)

* [Enhancement] Delete DS_Store file

* [Enhancement] Fix the build_segmentor
This commit is contained in:
Miao Zheng 2022-05-05 13:39:56 +08:00 committed by GitHub
parent 2d8be3712e
commit 2638bdf39b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -561,8 +561,7 @@
"datasets = [build_dataset(cfg.data.train)]\n",
"\n",
"# Build the detector\n",
"model = build_segmentor(\n",
" cfg.model, train_cfg=cfg.get('train_cfg'), test_cfg=cfg.get('test_cfg'))\n",
"model = build_segmentor(cfg.model)\n",
"# Add an attribute for visualization convenience\n",
"model.CLASSES = datasets[0].CLASSES\n",
"\n",