module TensorStream::Images

Public Class Methods

decode_jpeg(contents, channels: 0, dtype: :uint8, name: nil, new_shape: nil) click to toggle source
# File lib/tensor_stream/images.rb, line 10
def self.decode_jpeg(contents, channels: 0, dtype: :uint8, name: nil, new_shape: nil)
  _op(:decode_jpg, contents, channels: channels, data_type: dtype, name: name, new_shape: new_shape)
end
decode_png(contents, channels: 0, dtype: :uint8, name: nil, new_shape: nil) click to toggle source
# File lib/tensor_stream/images.rb, line 6
def self.decode_png(contents, channels: 0, dtype: :uint8, name: nil, new_shape: nil)
  _op(:decode_png, contents, channels: channels, data_type: dtype, name: name, new_shape: new_shape)
end
encode_png(contents, compression: -1, name: nil, new_shape: nil, resample_method: nil) click to toggle source
# File lib/tensor_stream/images.rb, line 14
def self.encode_png(contents, compression: -1, name: nil, new_shape: nil, resample_method: nil)
  check_allowed_types(contents, %i[uint8 uint16])
  contents = convert_to_tensor(contents, dtype: :uint16)
  _op(:encode_png, contents, compression: compression, name: name, new_shape: new_shape, resample_method: resample_method)
end