Skip to content

Commit

Permalink
Update Datumaro installation (#2715)
Browse files Browse the repository at this point in the history
* Update datumaro installation

* Update dm usage

* add no-binary to force setup.py invocation
  • Loading branch information
Maxim Zhiltsov authored Jan 26, 2021
1 parent d478f72 commit 473554b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
8 changes: 5 additions & 3 deletions cvat/apps/dataset_manager/formats/cvat.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
from cvat.apps.dataset_manager.util import make_zip_archive
from cvat.apps.engine.frame_provider import FrameProvider
from datumaro.components.extractor import DatasetItem
from datumaro.util.image import save_image

from .registry import exporter, importer

Expand Down Expand Up @@ -441,8 +440,11 @@ def load(file_object, annotations):
elif el.tag == 'image':
image_is_opened = True
frame_id = annotations.abs_frame_id(match_dm_item(
DatasetItem(id=el.attrib['id'], image=el.attrib['name']),
annotations))
DatasetItem(id=el.attrib['name'],
attributes={'frame': el.attrib['id']}
),
task_data=annotations
))
elif el.tag in supported_shapes and (track is not None or image_is_opened):
attributes = []
shape = {
Expand Down
2 changes: 1 addition & 1 deletion cvat/requirements/base.txt
Original file line number Diff line number Diff line change
Expand Up @@ -45,4 +45,4 @@ tensorflow==2.4.0 # Optional requirement of Datumaro
patool==1.12
diskcache==5.0.2
open3d==0.11.2
git+https://github.com/openvinotoolkit/datumaro@v0.1.4
datumaro==0.1.5.1 --no-binary=datumaro

0 comments on commit 473554b

Please sign in to comment.