在detectron2中,我们并不是只输入一个尺寸的图像
可以看到配置文件中:
# By default, {MIN,MAX}_SIZE options are used in transforms.ResizeShortestEdge.
# Please refer to ResizeShortestEdge for detailed definition.
# Size of the smallest side of the image during training
_C.INPUT.MIN_SIZE_TRAIN = (800,)
# Sample size of smallest side by choice or random selection from range give by
# INPUT.MIN_SIZE_TRAIN
_C.INPUT.MIN_SIZE_TRAIN_SAMPLING = "choice"
# Maximum size of the side of the image during training
_C.INPUT.MAX_SIZE_TRAIN = 1333
图像会经过裁剪的操作再送进模型,可以设置图像裁剪范围的区间,例如原配置中就是(800,1333)。
而最小输入尺寸,可以是个列表的形式,比如(800,900),这时候就会从(800,1333)和(900,1333)两个区间随机采样。