fix: build dictionary with ints instead of tensors

Former-commit-id: abbbeb13b852457450df03d926cd4689cbc464f6
javiribera · Dec 17, 2018 · b7152bc · b7152bc
1 parent 7536897
commit b7152bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/object-locator/locate.py b/object-locator/locate.py
@@ -290,7 +290,7 @@
 
         # Save a new line in the CSV corresonding to the resuls of this img
         res_dict = dictionaries[0]
-        res_dict['count'] = est_count
+        res_dict['count'] = est_count_int
         res_dict['locations'] = str(centroids_wrt_orig.tolist())
         for key, val in res_dict.copy().items():
             if 'height' in key or 'width' in key: