ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-24 16:26:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix TF binding to preserve previous behavior

Browse Source
This commit is contained in:
allegroai 2020-08-10 17:25:33 +03:00
parent e0ab86a64d
commit 4fbf0b9d4a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trains/binding/frameworks/tensorflow_bind.py
View File

@ -349,8 +349,7 @@ class EventTrainsWriter(object):
im = Image.open(output) im = Image.open(output)
image = np.asarray(im) image = np.asarray(im)
output.close() output.close()
if height is not None and width is not None: if height is not None and height > 0 and width is not None and width > 0:
assert height > 0 and width > 0, 'Image width and height should be positive integers'
# noinspection PyArgumentList # noinspection PyArgumentList
val = image.reshape(height, width, -1).astype(np.uint8) val = image.reshape(height, width, -1).astype(np.uint8)
else: else:
@ -760,6 +759,7 @@ class ProxyEventsWriter(object):
ret = None ret = None
for ev in self._events: for ev in self._events:
if hasattr(ev, '_get_sentinel_event'): if hasattr(ev, '_get_sentinel_event'):
# noinspection PyProtectedMember
ret = ev._get_sentinel_event() ret = ev._get_sentinel_event()
return ret return ret
@ -1270,7 +1270,7 @@ class PatchTensorFlowEager(object):
width = int(img_data_np[0].decode()) # noqa: F841 width = int(img_data_np[0].decode()) # noqa: F841
height = int(img_data_np[1].decode()) # noqa: F841 height = int(img_data_np[1].decode()) # noqa: F841
for i in range(2, img_data_np.size): for i in range(2, img_data_np.size):
img_data = {'width': -1, 'height': -1, img_data = {'width': None, 'height': None,
'colorspace': 'RGB', 'encodedImageString': img_data_np[i]} 'colorspace': 'RGB', 'encodedImageString': img_data_np[i]}
image_tag = str(tag) + '/sample_{}'.format(i - 2) if img_data_np.size > 3 else str(tag) image_tag = str(tag) + '/sample_{}'.format(i - 2) if img_data_np.size > 3 else str(tag)
event_writer._add_image(tag=image_tag, event_writer._add_image(tag=image_tag,