FPN_Pytorch
FPN_Pytorch copied to clipboard
训练0712的时候显存变大
@guoruoqian 请问为何在训练voc 0712数据集的时候显存占用会远远大于训练voc 07呢?我只能把batch size设为1才能训练。
因为训练voc 07 时,为了节省显存,会把图片crop(保留中心区域)进行训练。而voc 0712数据中,有一张图片的长宽比很大(2.+),而图片中的物体只出现在图片两侧,如果把图片crop保留中心位置,这张图片就没有物体了,训练会报错,所以在训练voc 0712数据时,不使用crop操作,没有节省现存的行为。