Commit 0b326566 authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent c3e27db1
......@@ -64,8 +64,8 @@ class Dataset(object):
boxes = extract_bboxes(masks)
boxes = torch.as_tensor(boxes, dtype=torch.float32)
# there is only one class
labels = torch.ones((num_objs,), dtype=torch.int64)
# two_classes
labels = torch.as_tensor(class_ids, dtype=torch.int64)
masks = torch.as_tensor(masks, dtype=torch.uint8)
image_id = torch.tensor([idx])
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment