update

foamliu · Oct 12, 2018 · 2efb928 · 2efb928
1 parent 7f25003
commit 2efb928
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -55,7 +55,7 @@
     outputs = outputs.cpu().numpy()
 
     for i in range(10):
-        result[i]['labels'] = outputs[i]
+        result[i]['labels'] = outputs[i].tolist()
 
     with open('result.json', 'w') as file:
         json.dump(result, file, indent=4)