oelin / semantic-shapes

Semantic shapes for deep neural networks.

semantic-shapes

Semantic shapes for deep neural networks.

Installation

pip install git+https://github.com/oelin/semantic-shapes

Usage

import semantic_shapes as ss

image_shape = ss.ImageShape(channels=3, height=256, width=256)

volume_shape = ss.VolumeShape(channels=3, depth=256, height=256, width=256)

video_shape = ss.VideoShape(duration=1024, channels=3, height=256, width=256)

audio_shape = ss.AudioShape(duration=1024, channels=32)

text_shape = ss.TextShape(length=1024)

sequence_shape = ss.SequenceShape(channels=256, length=1024)

vector_shape = ss.VectorShape(dimension=256)

matrix_shape = ss.MatrixShape(height=256, width=256)