diff --git a/new_tutorials/train/segmentation/fast_scnn.py b/new_tutorials/train/segmentation/fast_scnn.py index 53f1a528a090d6d4f278e47b54b2660dccde2e0d..9c48d31eda7b612243e65df124b51722c4ea59e4 100644 --- a/new_tutorials/train/segmentation/fast_scnn.py +++ b/new_tutorials/train/segmentation/fast_scnn.py @@ -35,7 +35,7 @@ eval_dataset = pdx.datasets.SegDataset( # 浏览器打开 https://0.0.0.0:8001即可 # 其中0.0.0.0为本机访问,如为远程服务, 改成相应机器IP -# https://paddlex.readthedocs.io/zh_CN/latest/apis/models/semantic_segmentation.html#hrnet +# https://paddlex.readthedocs.io/zh_CN/latest/apis/models/semantic_segmentation.html#fastscnn num_classes = len(train_dataset.labels) model = pdx.seg.FastSCNN(num_classes=num_classes) model.train(