package torch

  1. Overview
  2. Docs
On This Page
  1. Image writing
PyTorch bindings for OCaml

Install 

dune-project
 Dependency

github.com Readme Edit opam file Versions (18)

Authors

  1. L
    Laurent Mazare

Maintainers

  1. L
    Laurent Mazare <lmazare@gmail.com>

Sources

0.16.tar.gz
md5=8746b973b125b378d4f5f44800d1a0a6
sha512=86887d16a40f366e16a95f38863c4afd08ceb426b7a7341802451210739f5213322f0d10bb2f979adb26296c4fbee699a3ab5a1d264334f20c44bc05d6f8540e

doc/torch.vision/Torch_vision/Stb_image_write/index.html

Module Torch_vision.Stb_image_writeSource

Image writing

buffer simply is an alias to a bigarray with c_layout. Two kind of pixel buffers are manipulated:

  • int8 for images with 8-bit channels
  • float32 for images with floating point channels

Content of an image with c channels of width w and height h is represented as a contiguous sequence of items such that:

  • channels are interleaved
  • each pixel is made of c items
  • each line is made of w pixels
  • image is made of h lines
Sourcetype 'kind buffer = ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t constraint 'kind = ('a, 'b) Bigarray.kind
Sourceval png : string -> w:int -> h:int -> c:int -> int8 buffer -> unit
Sourceval bmp : string -> w:int -> h:int -> c:int -> int8 buffer -> unit
Sourceval tga : string -> w:int -> h:int -> c:int -> int8 buffer -> unit
Sourceval hdr : string -> w:int -> h:int -> c:int -> float32 buffer -> unit
Sourceval jpg : string -> w:int -> h:int -> c:int -> quality:int -> int8 buffer -> unit