from PIL import Image
from webdataset import TarWriter
from pathlib import Path


# 从磁盘读取图片
image_path = "/home/wst/dataset/tmp/20210106-152753_254_0__16115_1609918209412_0-1.jpg"
image = Image.open(image_path)

# 将图像转换为二进制格式
binary_data = image

# 如果PIL.Image格式要转换为cv2格式，用如下代码
# image = cv2.cvtColor(numpy.asarray(image),cv2.COLOR_RGB2BGR) 

# 把图片写入压缩包
writer = TarWriter("test.tar")
to_json = {
            "__key__": Path(image_path).stem,
            "jpg": binary_data
        }
writer.write(to_json)
writer.close()

import cv2
from webdataset import TarWriter
from pathlib import Path


# 从磁盘读取图片
image_path = "/home/wst/dataset/tmp/20210106-152753_254_0__16115_1609918209412_0-1.jpg"
image = cv2.imread(image_path)

# 将图像转换为PIL.Image格式，因为webdataset中的图片默认是这个格式。
binary_data = cv2.imencode('.jpg', image)[1].tobytes()

# 把图片写入压缩包
writer = TarWriter("test.tar")
to_json = {
            "__key__": Path(image_path).stem,
            "jpg": binary_data
        }
writer.write(to_json)
writer.close()

import PIL
assert isinstance(image, (PIL.Image.Image, np.ndarray)), type(image)