mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
[Enhancement] Fix the build_segmentor in colab demo (#1551)
* [Enhancement] Delete DS_Store file * [Enhancement] Fix the build_segmentor
This commit is contained in:
parent
2d8be3712e
commit
2638bdf39b
@ -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",
|
||||
|
Loading…
x
Reference in New Issue
Block a user