diff --git a/edgetpumodel.py b/edgetpumodel.py index 07f0425..86f8986 100644 --- 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 \ No newline at end of file + return det