ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-30 03:02:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Logger.report_image() throws warning

Browse Source
This commit is contained in:
allegroai 2021-09-20 15:52:06 +03:00
parent c9121debc2
commit 9966dd640c
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
clearml/backend_interface/metrics/events.py
View File

@ -229,7 +229,7 @@ class UploadEvent(MetricsEventAdapter):
self._key = None self._key = None
self._count = self._get_metric_count(metric, variant) self._count = self._get_metric_count(metric, variant)
if not file_history_size: if not file_history_size:
file_history_size = self._file_history_size file_history_size = int(self._file_history_size)
self._filename = kwargs.pop('override_filename', None) self._filename = kwargs.pop('override_filename', None)
if not self._filename: if not self._filename:
if file_history_size < 1: if file_history_size < 1:
@ -248,7 +248,7 @@ class UploadEvent(MetricsEventAdapter):
# e.g.: image.png -> .png or image.raw.gz -> .raw.gz # e.g.: image.png -> .png or image.raw.gz -> .raw.gz
filename_ext = kwargs.pop('override_filename_ext', None) filename_ext = kwargs.pop('override_filename_ext', None)
if filename_ext is None: if filename_ext is None:
filename_ext = self._format.lower() if self._image_data is not None else \ filename_ext = str(self._format).lower() if self._image_data is not None else \
'.' + '.'.join(pathlib2.Path(self._local_image_path).parts[-1].split('.')[1:]) '.' + '.'.join(pathlib2.Path(self._local_image_path).parts[-1].split('.')[1:])
# always add file extension to the uploaded target file # always add file extension to the uploaded target file
if filename_ext and filename_ext[0] != '.': if filename_ext and filename_ext[0] != '.':
@ -290,7 +290,7 @@ class UploadEvent(MetricsEventAdapter):
local_file = None local_file = None
# don't provide file in case this event is out of the history window # don't provide file in case this event is out of the history window
last_count = self._get_metric_count(self.metric, self.variant, next=False) last_count = self._get_metric_count(self.metric, self.variant, next=False)
if abs(self._count - last_count) > self._file_history_size: if abs(self._count - last_count) > int(self._file_history_size):
output = None output = None
elif isinstance(self._image_data, (six.StringIO, six.BytesIO)): elif isinstance(self._image_data, (six.StringIO, six.BytesIO)):
output = self._image_data output = self._image_data
@ -313,8 +313,8 @@ class UploadEvent(MetricsEventAdapter):
# serialize image # serialize image
image = Image.fromarray(image_data) image = Image.fromarray(image_data)
output = six.BytesIO() output = six.BytesIO()
image_format = Image.registered_extensions().get(self._format.lower(), 'JPEG') image_format = Image.registered_extensions().get(str(self._format).lower(), 'JPEG')
image.save(output, format=image_format, quality=self._quality) image.save(output, format=image_format, quality=int(self._quality))
output.seek(0) output.seek(0)
else: else:
# noinspection PyBroadException # noinspection PyBroadException