Torch.Vision

Documentation

data MNIST (m :: Type -> Type)

Constructors

MNIST

Fields

• batchSize :: Int
• mnistData :: MnistData

Instances

Applicative m => Dataset (m :: Type -> Type) (MNIST m) Int (Tensor, Tensor)

Defined in Torch.Vision

Methods

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

keys :: MNIST m -> Set Int

Monad m => Datastream m Int (MNIST m) (Tensor, Tensor)

Defined in Torch.Vision

Methods

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

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

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

grayScale10 :: String

grayScale70 :: [Char]

dispImage :: Tensor -> IO ()

data PixelFormat

Constructors

Y8

YF

YA8

RGB8

RGBF

RGBA8

YCbCr8

CMYK8

CMYK16

RGBA16

RGB16

Y16

YA16

Y32

Instances

Show PixelFormat

Defined in Torch.Vision

Methods

showsPrec :: Int -> PixelFormat -> ShowS

show :: PixelFormat -> String

showList :: [PixelFormat] -> ShowS

Eq PixelFormat

Defined in Torch.Vision

Methods

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

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

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

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

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

centerCrop :: Int -> Int -> Image PixelRGB8 -> Image PixelRGB8

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

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

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

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

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

pixelFormat :: DynamicImage -> PixelFormat

fromDynImage :: DynamicImage -> Tensor

fromImages :: [Image PixelRGB8] -> IO Tensor

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

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

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

hwc2chw :: Tensor -> Tensor

chw2hwc :: Tensor -> Tensor

randomIndexes :: Int -> [Int]