From a734d88d9af958da9b83fc1f193fa448f3d6cc94 Mon Sep 17 00:00:00 2001 From: Cathy Wong 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) -- Gitee