我正在使用KITTI的对象检测数据集通过ResNet 101训练一个更快的R-CNN,在MS COCO上进行了预训练。KITTI图像“大部分”尺寸为375x1242
当我有batch_size: 1
的时候,一切都很完美。我的keep_aspect_ratio_resizer
如下所示,因为它是由TensorFlow自己提出的。
min_dimension: 600
max_dimension: 1987
但是现在我想使用batch_size: 5,但是我一直收到尺寸不匹配的错误。因为有些图像的尺寸稍小,比如370x1224等。
我找不到一般的keep_aspect_ratio_resizer
值。我尝试了下面的值,基于我在错误消息中看到的值,但我真的不能使所有图像的大小都相同
min_dimension: 600
max_dimension: 1985
min_dimension: 599
max_dimension: 1985
https://stackoverflow.com/questions/55168271
复制相似问题