如何使用TensorFlow 2数据集从URL加载图像

import tensorflow as tf
import tensorflow_datasets as tfds
import matplotlib.pyplot as plt

def load_image_from_url(url):
    image = tf.image.decode_jpeg(tf.io.read_file(url), channels=3)
    image = tf.image.convert_image_dtype(image, tf.float32)
    return image

dataset, info = tfds.load('dataset_name', split='train', with_info=True)
fig = tfds.show_examples(dataset, info)
plt.show()

url = 'image_url'
image = load_image_from_url(url)
plt.imshow(image)
plt.axis('off')
plt.show()