Skip to content

Commit ed03158

Browse files
authored
fix regarding dataset name (#2001)
1 parent ecc3f91 commit ed03158

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

guides/ipynb/keras_hub/semantic_segmentation_deeplab_v3.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1053,7 +1053,7 @@
10531053
"test_ds = load(split=\"sbd_eval\")\n",
10541054
"test_ds = preprocess_inputs(test_ds)\n",
10551055
"\n",
1056-
"images, masks = next(iter(train_ds.take(1)))\n",
1056+
"images, masks = next(iter(test_ds.take(1)))\n",
10571057
"images = ops.convert_to_tensor(images)\n",
10581058
"masks = ops.convert_to_tensor(masks)\n",
10591059
"preds = ops.expand_dims(ops.argmax(model.predict(images), axis=-1), axis=-1)\n",

guides/keras_hub/semantic_segmentation_deeplab_v3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -827,7 +827,7 @@ def dict_to_tuple(x):
827827
test_ds = load(split="sbd_eval")
828828
test_ds = preprocess_inputs(test_ds)
829829

830-
images, masks = next(iter(train_ds.take(1)))
830+
images, masks = next(iter(test_ds.take(1)))
831831
images = ops.convert_to_tensor(images)
832832
masks = ops.convert_to_tensor(masks)
833833
preds = ops.expand_dims(ops.argmax(model.predict(images), axis=-1), axis=-1)

guides/md/keras_hub/semantic_segmentation_deeplab_v3.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -956,7 +956,7 @@ better accuracy and result train with more number of epochs.
956956
test_ds = load(split="sbd_eval")
957957
test_ds = preprocess_inputs(test_ds)
958958

959-
images, masks = next(iter(train_ds.take(1)))
959+
images, masks = next(iter(test_ds.take(1)))
960960
images = ops.convert_to_tensor(images)
961961
masks = ops.convert_to_tensor(masks)
962962
preds = ops.expand_dims(ops.argmax(model.predict(images), axis=-1), axis=-1)

0 commit comments

Comments
 (0)