Skip to content

Commit

Permalink
fix source code error in intermediate_source/torchvision_tutorial.rst (…
Browse files Browse the repository at this point in the history
…#793)

fix source code error in intermediate_source/torchvision_tutorial.rst
  • Loading branch information
yundaehyuck committed Nov 26, 2023
1 parent c5b2847 commit 825d72f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions intermediate_source/torchvision_tutorial.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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))
Expand Down

0 comments on commit 825d72f

Please sign in to comment.