From 53562582df8fe32943134671805dccae8d96af43 Mon Sep 17 00:00:00 2001 From: yundaehyuck Date: Mon, 16 Oct 2023 04:41:20 +0900 Subject: [PATCH] fix source code error in torchvision_tutorial.rst --- intermediate_source/torchvision_tutorial.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/intermediate_source/torchvision_tutorial.rst b/intermediate_source/torchvision_tutorial.rst index 763a06386..cd97cd028 100644 --- a/intermediate_source/torchvision_tutorial.rst +++ b/intermediate_source/torchvision_tutorial.rst @@ -142,7 +142,7 @@ PennFudan 데이터셋을 위한 코드를 작성해 보겠습니다. labels = torch.ones((num_objs,), dtype=torch.int64) masks = torch.as_tensor(masks, dtype=torch.uint8) - image_id = torch.tensor([idx]) + image_id = idx area = (boxes[:, 3] - boxes[:, 1]) * (boxes[:, 2] - boxes[:, 0]) # 모든 인스턴스는 군중(crowd) 상태가 아님을 가정합니다 iscrowd = torch.zeros((num_objs,), dtype=torch.int64) @@ -305,7 +305,7 @@ PennFudan 데이터셋을 위한 인스턴스 분할 모델 def get_transform(train): transforms = [] transforms.append(T.PILToTensor()) - transforms.append(T.ConvertImageDtype(torch.float)) + transforms.append(T.ToDtype(torch.float, scale = True)) if train: # (역자주: 학습시 50% 확률로 학습 영상을 좌우 반전 변환합니다) transforms.append(T.RandomHorizontalFlip(0.5))