Python: Return the vector rather than printing it.

This commit is contained in:
Brandon Amos 2015-09-24 17:30:33 -04:00
parent 122b06bd96
commit e808170324
1 changed files with 1 additions and 1 deletions

View File

@ -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(',')]