add subscript for fake image
This commit is contained in:
parent
402fa4f687
commit
0f2d6d2ec7
|
@ -29,6 +29,9 @@ class ONNXImage():
|
|||
self.source = source
|
||||
self.data = self
|
||||
|
||||
def __getitem__(self, *args):
|
||||
return torch.from_numpy(self.source.__getitem__(*args)).to(torch.float32)
|
||||
|
||||
def squeeze(self):
|
||||
self.source = np.squeeze(self.source, (0))
|
||||
return self
|
||||
|
|
Loading…
Reference in New Issue