if h > w:
     new_h, new_w = self.output_size * h / w, self.output_size
else:
     new_h, new_w = self.output_size, self.output_size * w / h    

  

相关文章: