From a734d88d9af958da9b83fc1f193fa448f3d6cc94 Mon Sep 17 00:00:00 2001
From: Cathy Wong <cathy.wong@huawei.com>
Date: Tue, 5 Jul 2022 13:31:52 -0400
Subject: [PATCH] fastscnn - correct doc api example
---
official/cv/fastscnn/src/dataloader.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/official/cv/fastscnn/src/dataloader.py b/official/cv/fastscnn/src/dataloader.py
index b6a770a63..27578e05e 100644
--- a/official/cv/fastscnn/src/dataloader.py
+++ b/official/cv/fastscnn/src/dataloader.py
@@ -40,8 +40,8 @@ class CitySegmentation(SegmentationDataset):
--------
>>> # Transforms for Normalization
>>> input_transform = transforms.Compose([
- >>> transforms.ToTensor(),
- >>> transforms.Normalize((.485, .456, .406), (.229, .224, .225)),
+ >>> CV.ToTensor(),
+ >>> CV.Normalize((.485, .456, .406), (.229, .224, .225), is_hwc=False),
>>> ])
>>> # Create Dataset
>>> trainset = CitySegmentation(split='train', transform=input_transform)
--
GitLab