|
|
@@ -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(
|