diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb index 4e2ed56e1e..066052612d 100644 --- a/demo/MMSegmentation_Tutorial.ipynb +++ b/demo/MMSegmentation_Tutorial.ipynb @@ -69,7 +69,7 @@ "# Install PyTorch\n", "!conda install pytorch=1.10.0 torchvision cudatoolkit=11.1 -c pytorch\n", "# Install MMCV\n", - "!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6/index.html" + "!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10/index.html" ] }, { @@ -156,8 +156,8 @@ }, "outputs": [], "source": [ - "config_file = '../configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py'\n", - "checkpoint_file = '../checkpoints/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth'" + "config_file = 'configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py'\n", + "checkpoint_file = 'checkpoints/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth'" ] }, { @@ -185,7 +185,7 @@ "outputs": [], "source": [ "# test a single image\n", - "img = './demo.png'\n", + "img = 'demo/demo.png'\n", "result = inference_segmentor(model, img)" ] },