ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix debug samples are not uploaded correctly (#924)

Browse Source 
* PR relating to bug #923.
---------

Co-authored-by: jday1 <jd685@cam.ac.uk>
Co-authored-by: jday1 <james.day@bglgroup.co.uk>
This commit is contained in:
jday1 2023-02-19 07:19:40 +00:00 committed by GitHub
parent e938f2800b
commit 541c2425e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/binding/frameworks/tensorflow_bind.py
View File

@ -398,11 +398,12 @@ class EventTrainsWriter(object):
with open(fd, "wb") as f: with open(fd, "wb") as f:
f.write(imdata) f.write(imdata)
return temp_file return temp_file
image = np.asarray(im)
output.close() output.close()
if height is not None and height > 0 and width is not None and width > 0: if height is not None and height > 0 and width is not None and width > 0:
val = np.array(im).reshape((height, width, -1)).astype(np.uint8) val = image.reshape((height, width, -1)).astype(np.uint8)
else: else:
val = np.array(im).astype(np.uint8) val = image.astype(np.uint8)
if val.ndim == 3 and val.shape[2] == 3: if val.ndim == 3 and val.shape[2] == 3:
if self._visualization_mode == 'BGR': if self._visualization_mode == 'BGR':
val = val[:, :, [2, 1, 0]] val = val[:, :, [2, 1, 0]]