Python: Return the vector rather than printing it.
This commit is contained in:
parent
122b06bd96
commit
e808170324
|
@ -27,4 +27,4 @@ class TorchWrap:
|
|||
|
||||
def forward(self, imgPath, timeout=10):
|
||||
self.p.stdin.write(imgPath+"\n")
|
||||
print([float(x) for x in self.p.stdout.readline().strip().split(',')])
|
||||
return [float(x) for x in self.p.stdout.readline().strip().split(',')]
|
||||
|
|
Loading…
Reference in New Issue