[Datumaro] Remote project export fixes (#1193)

* Export project with trask name

* Do not expose server paths

* Fix tfrecord mask reading in tf>1.14

* Setuptools compatibility
main
zhiltsov-max 6 years ago committed by GitHub
parent cd8ef2aca4
commit f2dfcc039b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -103,10 +103,12 @@ class cvat_rest_api_task_images(datumaro.SourceExtractor):
items = []
for entry in image_list:
item_id = entry['id']
item_filename = entry.get('name', str(item_id))
size = None
if entry.get('height') and entry.get('width'):
size = (entry['height'], entry['width'])
image = Image(data=self._make_image_loader(item_id), size=size)
image = Image(data=self._make_image_loader(item_id),
path=item_filename, size=size)
item = datumaro.DatasetItem(id=item_id, image=image)
items.append((item.id, item))

@ -94,7 +94,8 @@ class TaskProject:
CvatImagesDirExtractor)
def _import_from_task(self, user):
self._project = Project.generate(self._project_dir)
self._project = Project.generate(self._project_dir,
config={'project_name': self._db_task.name})
self._project.add_source('task_%s_images' % self._db_task.id, {
'url': self._db_task.get_data_dirname(),
@ -185,6 +186,7 @@ class TaskProject:
for db_image in self._db_task.image_set.all():
frame_info = {
'id': db_image.frame,
'name': osp.basename(db_image.path),
'width': db_image.width,
'height': db_image.height,
}

@ -148,7 +148,8 @@ class TfDetectionApiExtractor(SourceExtractor):
parsed_record['image/object/class/text'],
default_value=b'').numpy()
masks = tf.sparse.to_dense(
parsed_record['image/object/mask']).numpy()
parsed_record['image/object/mask'],
default_value=b'').numpy()
for label, label_id in zip(labels, label_ids):
label = label.decode('utf-8')

Loading…
Cancel
Save