Skip to content

Commit 825d72f

Browse files
authored
fix source code error in intermediate_source/torchvision_tutorial.rst (#793)
fix source code error in intermediate_source/torchvision_tutorial.rst
1 parent c5b2847 commit 825d72f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

โ€Žintermediate_source/torchvision_tutorial.rstโ€Ž

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ PennFudan ๋ฐ์ดํ„ฐ์…‹์„ ์œ„ํ•œ ์ฝ”๋“œ๋ฅผ ์ž‘์„ฑํ•ด ๋ณด๊ฒ ์Šต๋‹ˆ๋‹ค.
142142
labels = torch.ones((num_objs,), dtype=torch.int64)
143143
masks = torch.as_tensor(masks, dtype=torch.uint8)
144144
145-
image_id = torch.tensor([idx])
145+
image_id = idx
146146
area = (boxes[:, 3] - boxes[:, 1]) * (boxes[:, 2] - boxes[:, 0])
147147
# ๋ชจ๋“  ์ธ์Šคํ„ด์Šค๋Š” ๊ตฐ์ค‘(crowd) ์ƒํƒœ๊ฐ€ ์•„๋‹˜์„ ๊ฐ€์ •ํ•ฉ๋‹ˆ๋‹ค
148148
iscrowd = torch.zeros((num_objs,), dtype=torch.int64)
@@ -305,7 +305,7 @@ PennFudan ๋ฐ์ดํ„ฐ์…‹์„ ์œ„ํ•œ ์ธ์Šคํ„ด์Šค ๋ถ„ํ•  ๋ชจ๋ธ
305305
def get_transform(train):
306306
transforms = []
307307
transforms.append(T.PILToTensor())
308-
transforms.append(T.ConvertImageDtype(torch.float))
308+
transforms.append(T.ToDtype(torch.float, scale = True))
309309
if train:
310310
# (์—ญ์ž์ฃผ: ํ•™์Šต์‹œ 50% ํ™•๋ฅ ๋กœ ํ•™์Šต ์˜์ƒ์„ ์ขŒ์šฐ ๋ฐ˜์ „ ๋ณ€ํ™˜ํ•ฉ๋‹ˆ๋‹ค)
311311
transforms.append(T.RandomHorizontalFlip(0.5))

0 commit comments

Comments
ย (0)