def downscale(self, images, scale):
   if self._scale_mode == 'ALL':
     return layers.downscale(images, scale)
   elif self._scale_mode == 'H':
     return layers.downscale_height(images, scale)
   else:
     self._raise_unsupported_scale_mode_error()
예제 #2
0
파일: networks.py 프로젝트: adarob/magenta
 def downscale(self, images, scale):
   if self._scale_mode == 'ALL':
     return layers.downscale(images, scale)
   elif self._scale_mode == 'H':
     return layers.downscale_height(images, scale)
   else:
     self._raise_unsupported_scale_mode_error()