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

Update MMSegmentation_Tutorial.ipynb (#1366)

Browse Source 
typos
This commit is contained in:
andife 2022-03-10 13:07:12 +01:00 committed by GitHub
parent 054dc66145
commit d06a84d54d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
demo/MMSegmentation_Tutorial.ipynb
View File

@ -369,7 +369,7 @@
"id": "HchvmGYB_rrO" "id": "HchvmGYB_rrO"
}, },
"source": [ "source": [
"After downloading the data, we need to implement `load_annotations` function in the new dataset class `StandfordBackgroundDataset`." "After downloading the data, we need to implement `load_annotations` function in the new dataset class `StanfordBackgroundDataset`."
] ]
}, },
{ {
@ -384,7 +384,7 @@
"from mmseg.datasets.custom import CustomDataset\n", "from mmseg.datasets.custom import CustomDataset\n",
"\n", "\n",
"@DATASETS.register_module()\n", "@DATASETS.register_module()\n",
"class StandfordBackgroundDataset(CustomDataset):\n", "class StanfordBackgroundDataset(CustomDataset):\n",
" CLASSES = classes\n", " CLASSES = classes\n",
" PALETTE = palette\n", " PALETTE = palette\n",
" def __init__(self, split, **kwargs):\n", " def __init__(self, split, **kwargs):\n",
@ -440,7 +440,7 @@
"source": [ "source": [
"from mmseg.apis import set_random_seed\n", "from mmseg.apis import set_random_seed\n",
"\n", "\n",
"# Since we use ony one GPU, BN is used instead of SyncBN\n", "# Since we use only one GPU, BN is used instead of SyncBN\n",
"cfg.norm_cfg = dict(type='BN', requires_grad=True)\n", "cfg.norm_cfg = dict(type='BN', requires_grad=True)\n",
"cfg.model.backbone.norm_cfg = cfg.norm_cfg\n", "cfg.model.backbone.norm_cfg = cfg.norm_cfg\n",
"cfg.model.decode_head.norm_cfg = cfg.norm_cfg\n", "cfg.model.decode_head.norm_cfg = cfg.norm_cfg\n",
@ -450,7 +450,7 @@
"cfg.model.auxiliary_head.num_classes = 8\n", "cfg.model.auxiliary_head.num_classes = 8\n",
"\n", "\n",
"# Modify dataset type and path\n", "# Modify dataset type and path\n",
"cfg.dataset_type = 'StandfordBackgroundDataset'\n", "cfg.dataset_type = 'StanfordBackgroundDataset'\n",
"cfg.data_root = data_root\n", "cfg.data_root = data_root\n",
"\n", "\n",
"cfg.data.samples_per_gpu = 8\n", "cfg.data.samples_per_gpu = 8\n",