Python LabelToLongTensorの例

コード例 #1

ファイル: main_runfile.py プロジェクト: KateZi/agame-vos

 def label_read(path):
     if (os.path.exists(path)):
         pic = Image.open(path)
         transform = tv.transforms.Compose([LabelToLongTensor()])
         label = transform(pic)
     else:
         label = torch.LongTensor(1, *size).fill_(
             255)  # Put label that will be ignored
     return label

コード例 #2

ファイル: davis_runfile.py プロジェクト: zechendev/agame-vos

 def label_read(path):
     if os.path.exists(path):
         pic = Image.open(path)
         transform = tv.transforms.Compose(
             [tv.transforms.Resize(size, interpolation=Image.NEAREST),
              LabelToLongTensor()])
         label = transform(pic)
     else:
         label = torch.LongTensor(1,*size).fill_(255) # Put label that will be ignored
     return label

コード例 #3

ファイル: test.py プロジェクト: CV-IP/PMVOS

 def label_read(path):
     pic = Image.open(path)
     transform = tv.transforms.Compose([LabelToLongTensor()])
     label = transform(pic)
     return label