Added drop_img_dim for rescale_img.