我正在尝试训练我的模型ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.tar.gz，不幸的是，这就是我得到的
有人对此有解决方案吗？
InvalidArgumentError: Graph execution error:
image_size must contain 3 elements[4]
[[{{node RandomCropImage/sample_distorted_bounding_box/SampleDistortedBoundingBoxV2}}]]
[[MultiDeviceIteratorGetNextFromShard]]
[[RemoteCall]]
[[while/body/_1/IteratorGetNext]] [Op:__inference__dist_train_step_51958]
该问题很可能与非RGB图像有关。在加载任何图像之前，请确保将它们转换为RGB(3个通道(。
请使用以下代码查看哪个图像不处于RGB模式并将其删除。
from PIL import Image
import os
path = 'PATH TO THE IMAGES'
for file in os.listdir(path):
extension = file.split('.')[-1]
# image extension could be png, jpg, etc
if extension == 'jpg':
filepath = path+file
img = Image.open(filepath)
if img.mode != 'RGB':
print(file+', '+img.mode)