Merge pull request #10 from mrl-amrl/yolo-format-filename

fix(yoloannotation): just last file extension be replaced with txt no…

preimutils/object_detection/yolo/coco2yolo.py

@@ -79,7 +79,7 @@ def coco2yolo(self):
     def _save_txt(self, anno_dict):
         for k, v in anno_dict.items():
             ext_name = v[0][0].split(".")[-1]
-            file_name = v[0][0].replace(ext_name, 'txt')
+            file_name = v[0][0][:-len(ext_name)] + 'txt'
             with open(os.path.join(self.output, file_name), 'w', encoding='utf-8') as f:
                 for obj in v:
                     cat_name = self.coco_id_name_map.get(obj[1])