Ejemplos de LabelToLongTensor en Python

Ejemplo n.º 1

Archivo: main_runfile.py Proyecto: 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

Ejemplo n.º 2

Archivo: davis_runfile.py Proyecto: 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

Ejemplo n.º 3

Archivo: test.py Proyecto: CV-IP/PMVOS

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