From 6d3c9690febc2b8e5cdb05c4031d066d40df3b1b Mon Sep 17 00:00:00 2001 From: Eike Gebauer Date: Mon, 23 Jun 2025 12:31:23 +0200 Subject: [PATCH] Added imghdr without the deprecation warning to replace missing standard lib module. --- tensorboard/plugins/image/images_plugin.py | 2 +- tensorboard/plugins/metrics/metrics_plugin.py | 2 +- .../plugins/projector/projector_plugin.py | 3 +- tensorboard/util/imghdr.py | 175 ++++++++++++++++++ 4 files changed, 178 insertions(+), 4 deletions(-) create mode 100644 tensorboard/util/imghdr.py diff --git a/tensorboard/plugins/image/images_plugin.py b/tensorboard/plugins/image/images_plugin.py index 15999c21c4..8d38309de4 100644 --- a/tensorboard/plugins/image/images_plugin.py +++ b/tensorboard/plugins/image/images_plugin.py @@ -15,7 +15,6 @@ """The TensorBoard Images plugin.""" -import imghdr import urllib.parse from werkzeug import wrappers @@ -26,6 +25,7 @@ from tensorboard.data import provider from tensorboard.plugins import base_plugin from tensorboard.plugins.image import metadata +from tensorboard.util import imghdr _IMGHDR_TO_MIMETYPE = { diff --git a/tensorboard/plugins/metrics/metrics_plugin.py b/tensorboard/plugins/metrics/metrics_plugin.py index e9c97c866c..c0bbb6995c 100644 --- a/tensorboard/plugins/metrics/metrics_plugin.py +++ b/tensorboard/plugins/metrics/metrics_plugin.py @@ -16,7 +16,6 @@ import collections -import imghdr import json from werkzeug import wrappers @@ -30,6 +29,7 @@ from tensorboard.plugins.image import metadata as image_metadata from tensorboard.plugins.metrics import metadata from tensorboard.plugins.scalar import metadata as scalar_metadata +from tensorboard.util import imghdr _IMGHDR_TO_MIMETYPE = { diff --git a/tensorboard/plugins/projector/projector_plugin.py b/tensorboard/plugins/projector/projector_plugin.py index bdbe70c759..7565e17ccd 100644 --- a/tensorboard/plugins/projector/projector_plugin.py +++ b/tensorboard/plugins/projector/projector_plugin.py @@ -17,7 +17,6 @@ import collections import functools -import imghdr import mimetypes import os import threading @@ -35,7 +34,7 @@ from tensorboard.plugins import base_plugin from tensorboard.plugins.projector import metadata from tensorboard.plugins.projector.projector_config_pb2 import ProjectorConfig -from tensorboard.util import tb_logging +from tensorboard.util import tb_logging, imghdr logger = tb_logging.get_logger() diff --git a/tensorboard/util/imghdr.py b/tensorboard/util/imghdr.py new file mode 100644 index 0000000000..1e3b3da073 --- /dev/null +++ b/tensorboard/util/imghdr.py @@ -0,0 +1,175 @@ +"""Recognize image file formats based on their first few bytes.""" + +from os import PathLike + +__all__ = ["what"] + +#-------------------------# +# Recognize image headers # +#-------------------------# + +def what(file, h=None): + """Return the type of image contained in a file or byte stream.""" + f = None + try: + if h is None: + if isinstance(file, (str, PathLike)): + f = open(file, 'rb') + h = f.read(32) + else: + location = file.tell() + h = file.read(32) + file.seek(location) + for tf in tests: + res = tf(h, f) + if res: + return res + finally: + if f: f.close() + return None + + +#---------------------------------# +# Subroutines per image file type # +#---------------------------------# + +tests = [] + +def test_jpeg(h, f): + """Test for JPEG data with JFIF or Exif markers; and raw JPEG.""" + if h[6:10] in (b'JFIF', b'Exif'): + return 'jpeg' + elif h[:4] == b'\xff\xd8\xff\xdb': + return 'jpeg' + +tests.append(test_jpeg) + +def test_png(h, f): + """Verify if the image is a PNG.""" + if h.startswith(b'\211PNG\r\n\032\n'): + return 'png' + +tests.append(test_png) + +def test_gif(h, f): + """Verify if the image is a GIF ('87 or '89 variants).""" + if h[:6] in (b'GIF87a', b'GIF89a'): + return 'gif' + +tests.append(test_gif) + +def test_tiff(h, f): + """Verify if the image is a TIFF (can be in Motorola or Intel byte order).""" + if h[:2] in (b'MM', b'II'): + return 'tiff' + +tests.append(test_tiff) + +def test_rgb(h, f): + """test for the SGI image library.""" + if h.startswith(b'\001\332'): + return 'rgb' + +tests.append(test_rgb) + +def test_pbm(h, f): + """Verify if the image is a PBM (portable bitmap).""" + if len(h) >= 3 and \ + h[0] == ord(b'P') and h[1] in b'14' and h[2] in b' \t\n\r': + return 'pbm' + +tests.append(test_pbm) + +def test_pgm(h, f): + """Verify if the image is a PGM (portable graymap).""" + if len(h) >= 3 and \ + h[0] == ord(b'P') and h[1] in b'25' and h[2] in b' \t\n\r': + return 'pgm' + +tests.append(test_pgm) + +def test_ppm(h, f): + """Verify if the image is a PPM (portable pixmap).""" + if len(h) >= 3 and \ + h[0] == ord(b'P') and h[1] in b'36' and h[2] in b' \t\n\r': + return 'ppm' + +tests.append(test_ppm) + +def test_rast(h, f): + """test for the Sun raster file.""" + if h.startswith(b'\x59\xA6\x6A\x95'): + return 'rast' + +tests.append(test_rast) + +def test_xbm(h, f): + """Verify if the image is a X bitmap (X10 or X11).""" + if h.startswith(b'#define '): + return 'xbm' + +tests.append(test_xbm) + +def test_bmp(h, f): + """Verify if the image is a BMP file.""" + if h.startswith(b'BM'): + return 'bmp' + +tests.append(test_bmp) + +def test_webp(h, f): + """Verify if the image is a WebP.""" + if h.startswith(b'RIFF') and h[8:12] == b'WEBP': + return 'webp' + +tests.append(test_webp) + +def test_exr(h, f): + """verify is the image ia a OpenEXR fileOpenEXR.""" + if h.startswith(b'\x76\x2f\x31\x01'): + return 'exr' + +tests.append(test_exr) + +#--------------------# +# Small test program # +#--------------------# + +def test(): + import sys + recursive = 0 + if sys.argv[1:] and sys.argv[1] == '-r': + del sys.argv[1:2] + recursive = 1 + try: + if sys.argv[1:]: + testall(sys.argv[1:], recursive, 1) + else: + testall(['.'], recursive, 1) + except KeyboardInterrupt: + sys.stderr.write('\n[Interrupted]\n') + sys.exit(1) + +def testall(list, recursive, toplevel): + import sys + import os + for filename in list: + if os.path.isdir(filename): + print(filename + '/:', end=' ') + if recursive or toplevel: + print('recursing down:') + import glob + names = glob.glob(os.path.join(glob.escape(filename), '*')) + testall(names, recursive, 0) + else: + print('*** directory (use -r) ***') + else: + print(filename + ':', end=' ') + sys.stdout.flush() + try: + print(what(filename)) + except OSError: + print('*** not found ***') + +if __name__ == '__main__': + test()