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.4.tar.gz
md5=9547e9e025dacd52e405ff699539c582
sha512=23fd9bef6f5f11c55171f2383a2f7ca57330511af6521a4579410e002d8667a91e764aecc2deb1cf8d7bf3b0e988cd3020850fa4a5a1ec713dfb110ec7352892

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