问题描述:
set_dynamic_columns(columns=None)
【操作步骤&问题现象】
1、在训练mask_rcnn时,由于图片尺寸过大以及存在单个图片内目标过多( >1000),所以不能使用pad_to_max的操作使得在生成数据集时保持每个row的shape是相同的。在官方文档中看到可以设置动态的shape信息(set_dynamic_columns(columns=None) ),但是我需要的是output_columns的shape不一致,该如何使用 set_dynamic_columns呢?
【截图信息】
尝试的方法:
报错信息
解答:
根据您的error日志,是batch维度不一致产生的报错,设置set_dynamic_shape并不能帮助你batch不同维度的数据,batch是不能把不同shape的数据batch到一起的,要么就不用batch,要么就把将同一个batch达到一致。