我正在尝试使用 tensorflow 脚本重新训练 ssd_mobilenet_v2： https ://github.com/tensorflow/models/blob/master/research/object_detection/model_main_tf2.py 和默认配置文件： https ://github.com/tensorflow /models/blob/master/research/object_detection/configs/tf2/ssd_mobilenet_v2_320x320_coco17_tpu-8.config
但我想在拜耳图像上训练网络（这些图像只有 1 个通道，就像黑白图像一样）。
我不知道要编辑什么以将输入大小从（320x320x3）更改为（320x320x1）。如果有人能给我一些指示，那就太好了。