add image path in load annotations for cvat format

5 years ago · aad580b9e0
parent 43fe33ff9f
commit aad580b9e0
1 changed files with 2 additions and 1 deletions
--- a/cvat/apps/dataset_manager/formats/cvat.py
+++ b/cvat/apps/dataset_manager/formats/cvat.py
@ -442,7 +442,8 @@ def load(file_object, annotations):
                image_is_opened = True
                frame_id = annotations.abs_frame_id(match_dm_item(
                    DatasetItem(id=osp.splitext(el.attrib['name'])[0],
-                        attributes={'frame': el.attrib['id']}
+                        attributes={'frame': el.attrib['id']},
+                        image=el.attrib['name']
                    ),
                    task_data=annotations
                ))