mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
* [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
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
demo/MMSegmentation_Tutorial.ipynb
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",