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.11.tar.gz
md5=45ccaed01b4aea5d1f9ae0c89c85f78b
sha512=c3d02319adb88f641c0eaf312c4a15525a28f1ee8ef46be10111aafbefb40b29fc1746733be49f6a2049a5a17ccc203a8c0481a18f70eed05b9cf4a34830a2c2

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