Merge pull request jveitchmichaelis#14 from fvalle1/main

Fix txt extension
sebastianfis · Dec 17, 2023 · 422d209 · 422d209
2 parents 784d9be + dcee6db
commit 422d209
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/edgetpumodel.py b/edgetpumodel.py
@@ -261,10 +261,10 @@ def process_predictions(self, det, output_image, pad, output_path="detection.jpg
                     output[base]['cls_name'] = self.names[c]
 
             if save_txt:
-                output_txt = base+"txt"
+                output_txt = base+".txt"
                 with open(output_txt, 'w') as f:
                    json.dump(output, f, indent=1)
             if save_img:
               cv2.imwrite(output_path, output_image)
 
-        return det
+        return det