hasktorch-0.2.0.0: Functional differentiable programming in Haskell
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Vision

Documentation

data MNIST (m :: * -> *) Source #

Constructors

MNIST 

Fields

Instances

Instances details
Applicative m => Dataset (m :: Type -> Type) (MNIST m) Int (Tensor, Tensor) Source # 
Instance details

Defined in Torch.Vision

Methods

getItem :: MNIST m -> Int -> m (Tensor, Tensor) Source #

keys :: MNIST m -> Set Int Source #

Monad m => Datastream m Int (MNIST m) (Tensor, Tensor) Source # 
Instance details

Defined in Torch.Vision

Methods

streamSamples :: MNIST m -> Int -> ListT m (Tensor, Tensor) Source #

getLabels' :: Int -> MnistData -> [Int] -> Tensor Source #

getImages' :: Int -> Int -> MnistData -> [Int] -> Tensor Source #

grayScale10 :: String Source #

grayScale70 :: [Char] Source #

dispImage :: Tensor -> IO () Source #

data PixelFormat Source #

Constructors

Y8 
YF 
YA8 
RGB8 
RGBF 
RGBA8 
YCbCr8 
CMYK8 
CMYK16 
RGBA16 
RGB16 
Y16 
YA16 
Y32 

Instances

Instances details
Show PixelFormat Source # 
Instance details

Defined in Torch.Vision

Methods

showsPrec :: Int -> PixelFormat -> ShowS Source #

show :: PixelFormat -> String Source #

showList :: [PixelFormat] -> ShowS Source #

Eq PixelFormat Source # 
Instance details

Defined in Torch.Vision

Methods

(==) :: PixelFormat -> PixelFormat -> Bool Source #

(/=) :: PixelFormat -> PixelFormat -> Bool Source #

readImage :: FilePath -> IO (Either String (Tensor, PixelFormat)) Source #

readImageAsRGB8 :: FilePath -> IO (Either String Tensor) Source #

readImageAsRGB8WithScaling :: FilePath -> Int -> Int -> Bool -> IO (Either String (Image PixelRGB8, Tensor)) Source #

centerCrop :: Int -> Int -> Image PixelRGB8 -> Image PixelRGB8 Source #

drawLine :: Int -> Int -> Int -> Int -> (Int, Int, Int) -> Image PixelRGB8 -> IO () Source #

drawRect :: Int -> Int -> Int -> Int -> (Int, Int, Int) -> Image PixelRGB8 -> IO () Source #

drawString :: String -> Int -> Int -> (Int, Int, Int) -> (Int, Int, Int) -> Image PixelRGB8 -> IO () Source #

drawChar :: Int -> Int -> Int -> (Int, Int, Int) -> (Int, Int, Int) -> Image PixelRGB8 -> IO () Source #

resizeRGB8 :: Int -> Int -> Bool -> Image PixelRGB8 -> Image PixelRGB8 Source #

pixelFormat :: DynamicImage -> PixelFormat Source #

fromDynImage :: DynamicImage -> Tensor Source #

fromImages :: [Image PixelRGB8] -> IO Tensor Source #

writeImage :: forall p. Pixel p => Int -> Int -> Int -> p -> Tensor -> IO (Image p) Source #

writeBitmap :: FilePath -> Tensor -> IO () Source #

writePng :: FilePath -> Tensor -> IO () Source #

hwc2chw :: Tensor -> Tensor Source #

chw2hwc :: Tensor -> Tensor Source #

randomIndexes :: Int -> [Int] Source #