Index A | B | C | D | E | F | G | H | I | L | M | N | O | P | R | S | T | U | V | W | X | Y | Z A accept() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedArrayTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedStructTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.PlainFieldTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.SchemaContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructFieldContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.UnquotedIdentifierContext method) apply_model_spec() (in module rikai.testing.utils) area (rikai.types.geometry.Box2d property) area() (in module rikai.spark.functions.geometry) ARRAY (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) array() (in module rikai.numpy) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) ARRAY() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayTypeContext method) array() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedArrayTypeContext method) assert_count_equal() (in module rikai.testing.asserters) Asset (class in rikai.mixin) atn (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) B BaseResolver (class in rikai.parquet.resolver) Box2d (class in rikai.types.geometry) box2d() (in module rikai.spark.functions.geometry) box2d_from_center() (in module rikai.spark.functions.geometry) box2d_from_top_left() (in module rikai.spark.functions.geometry) Box2dType (class in rikai.spark.types) (class in rikai.spark.types.geometry) Box3d (class in rikai.types.geometry) Box3dType (class in rikai.spark.types) (class in rikai.spark.types.geometry) buildParseTrees (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) C CaseChangingStream (class in rikai.spark.sql.schema) center (rikai.types.geometry.Box3d attribute) channelNames (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) Classification (class in rikai.sklearn.models) ClassificationModelType (class in rikai.pytorch.models.torch) COCO_RLE (rikai.types.geometry.Mask.Type attribute) codegen() (rikai.spark.sql.codegen.testing.TestModel method) ColumnNotFoundError command_from_spec() (in module rikai.spark.sql.codegen) convert() (in module rikai.contrib.datasets.coco) copy() (in module rikai.io) (in module rikai.spark.functions.io) copyFrom() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.IdentifierContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructContext method) count() (rikai.parquet.dataset.Dataset method) crop() (in module rikai.spark.functions.vision) (rikai.types.vision.Image method) D data (rikai.mixin.Asset attribute) Dataset (class in rikai.parquet.dataset) (class in rikai.pytorch.data) (class in rikai.pytorch.vision) dataType() (rikai.spark.sql.model.ModelType method) decisionsToDFA (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) decode() (in module rikai.types.rle) DEFAULT_MIN_SCORE (rikai.pytorch.models.ssd_class_scores.SSDClassScoresModelType attribute) DEFAULT_SCHEME (rikai.parquet.resolver.Resolver attribute) DefaultResolver (class in rikai.parquet.resolver) deserialize() (rikai.spark.types.Box2dType method) (rikai.spark.types.Box3dType method) (rikai.spark.types.geometry.Box2dType method) (rikai.spark.types.geometry.Box3dType method) (rikai.spark.types.geometry.MaskType method) (rikai.spark.types.geometry.PointType method) (rikai.spark.types.ImageType method) (rikai.spark.types.MaskType method) (rikai.spark.types.NDArrayType method) (rikai.spark.types.PointType method) (rikai.spark.types.SegmentType method) (rikai.spark.types.video.VideoStreamType method) (rikai.spark.types.video.YouTubeVideoType method) (rikai.spark.types.VideoStreamType method) (rikai.spark.types.vision.ImageType method) (rikai.spark.types.YouTubeVideoType method) df_to_rikai() (in module rikai.parquet.writer) (in module rikai.spark.utils) DimensionalityReduction (class in rikai.sklearn.models) display() (rikai.mixin.Displayable method) (rikai.types.video.VideoStream method) (rikai.types.video.YouTubeVideo method) (rikai.types.vision.Image method) Displayable (class in rikai.mixin) draw() (rikai.types.vision.Image method) Drawable (class in rikai.mixin) DummyModelSpec (class in rikai.spark.sql.codegen.dummy) DummyRegistry (class in rikai.spark.sql.codegen.dummy) E empty() (in module rikai.numpy) encode() (in module rikai.types.rle) EOF (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) eval() (rikai.pytorch.models.feature_extractor.FeatureExtractor method) exists() (in module rikai.io) F FeatureExtractor (class in rikai.pytorch.models.feature_extractor) FeatureExtractorType (class in rikai.pytorch.models.feature_extractor) field() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructTypeContext method) fieldType() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.SchemaContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructFieldContext method) FileSystemRegistry (class in rikai.spark.sql.codegen.fs) find_class() (in module rikai.internal.reflection) find_func() (in module rikai.internal.reflection) flavor (rikai.spark.sql.model.ModelSpec property) forward() (rikai.pytorch.models.feature_extractor.FeatureExtractor method) (rikai.pytorch.models.ssd_class_scores.SSDClassScoresExtractor method) from_array() (rikai.types.vision.Image class method) from_center() (rikai.types.geometry.Box2d class method) from_coco_rle() (rikai.types.geometry.Mask static method) from_mask() (rikai.types.geometry.Mask static method) from_pil() (rikai.types.vision.Image static method) from_polygon() (rikai.types.geometry.Mask static method) from_rle() (rikai.types.geometry.Mask static method) from_top_left() (rikai.types.geometry.Box2d class method) G generate_inference_func() (in module rikai.spark.sql.codegen.pytorch) generate_udf() (in module rikai.spark.sql.codegen.pytorch) (in module rikai.spark.sql.codegen.sklearn) get_default_jar_version() (in module rikai.spark.utils) get_schema() (rikai.parquet.resolver.BaseResolver method) (rikai.parquet.resolver.DefaultResolver method) (rikai.parquet.resolver.Resolver class method) get_stream() (rikai.types.video.YouTubeVideo method) getRuleIndex() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.IdentifierContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.SchemaContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructContext method) grammarFileName (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) H has_func() (in module rikai.internal.reflection) heading (rikai.types.geometry.Box3d attribute) height (rikai.types.geometry.Box2d property) (rikai.types.geometry.Box3d attribute) I IDENTIFIER (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) identifier() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.PlainFieldTypeContext method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructFieldContext method) IDENTIFIER() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.UnquotedIdentifierContext method) Image (class in rikai.types.vision) image_copy() (in module rikai.spark.functions.vision) ImageType (class in rikai.spark.types) (class in rikai.spark.types.vision) init() (in module rikai.spark.functions) init_spark_session() (in module rikai.spark.utils) iou() (rikai.types.geometry.Box2d method) (rikai.types.geometry.Mask method) ious() (rikai.types.geometry.Box2d static method) is_embedded (rikai.mixin.Asset property) L LA() (rikai.spark.sql.schema.CaseChangingStream method) label_fn() (in module rikai.pytorch.models.keypointrcnn) length (rikai.types.geometry.Box3d attribute) literalNames (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) load_label_fn() (rikai.spark.sql.model.ModelSpec method) load_model() (rikai.pytorch.models.feature_extractor.FeatureExtractorType method) (rikai.pytorch.models.ssd_class_scores.SSDClassScoresModelType method) (rikai.pytorch.models.torch.TorchModelType method) (rikai.sklearn.models.SklearnModelType method) (rikai.spark.sql.codegen.dummy.DummyModelSpec method) (rikai.spark.sql.model.ModelSpec method) (rikai.spark.sql.model.ModelType method) (rikai.tensorflow.models.TensorflowModelType method) load_model_from_uri() (in module rikai.spark.sql.codegen.pytorch) log_model() (rikai.spark.sql.codegen.mlflow_logger.MlflowLogger method) M make_model_spec() (rikai.spark.sql.codegen.base.Registry method) (rikai.spark.sql.codegen.dummy.DummyRegistry method) (rikai.spark.sql.codegen.fs.FileSystemRegistry method) Mask (class in rikai.types.geometry) Mask.Type (class in rikai.types.geometry) MaskType (class in rikai.spark.types) (class in rikai.spark.types.geometry) metadata (rikai.parquet.dataset.Dataset property) MlflowLogger (class in rikai.spark.sql.codegen.mlflow_logger) model_type (rikai.spark.sql.model.ModelSpec property) model_uri (rikai.spark.sql.model.ModelSpec property) ModelSpec (class in rikai.spark.sql.model) ModelType (class in rikai.spark.sql.model) modeNames (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) module rikai rikai.conf rikai.contrib rikai.contrib.datasets rikai.contrib.datasets.coco rikai.contrib.tfhub rikai.contrib.tfhub.tensorflow rikai.contrib.tfhub.tensorflow.ssd rikai.contrib.torch rikai.contrib.torch.detections rikai.contrib.torch.transforms rikai.contrib.torch.transforms.fasterrcnn_resnet50_fpn rikai.contrib.torch.transforms.ssd rikai.contrib.torchhub rikai.contrib.torchhub.pytorch rikai.contrib.torchhub.pytorch.vision rikai.contrib.torchhub.pytorch.vision.resnet rikai.contrib.video rikai.contrib.video.functions rikai.exceptions rikai.internal rikai.internal.reflection rikai.internal.uri_utils rikai.io rikai.logging rikai.mixin rikai.numpy rikai.parquet rikai.parquet.dataset rikai.parquet.resolver rikai.parquet.writer rikai.pytorch rikai.pytorch.data rikai.pytorch.models rikai.pytorch.models.convnext rikai.pytorch.models.efficientnet rikai.pytorch.models.fasterrcnn rikai.pytorch.models.feature_extractor rikai.pytorch.models.keypointrcnn rikai.pytorch.models.maskrcnn rikai.pytorch.models.resnet rikai.pytorch.models.retinanet rikai.pytorch.models.ssd rikai.pytorch.models.ssd_class_scores rikai.pytorch.models.torch rikai.pytorch.pandas rikai.pytorch.transforms rikai.pytorch.vision rikai.sklearn rikai.sklearn.models rikai.spark rikai.spark.functions rikai.spark.functions.geometry rikai.spark.functions.io rikai.spark.functions.vision rikai.spark.sql rikai.spark.sql.codegen rikai.spark.sql.codegen.base rikai.spark.sql.codegen.dummy rikai.spark.sql.codegen.fs rikai.spark.sql.codegen.mlflow_logger rikai.spark.sql.codegen.pytorch rikai.spark.sql.codegen.sklearn rikai.spark.sql.codegen.testing rikai.spark.sql.exceptions rikai.spark.sql.generated rikai.spark.sql.generated.RikaiModelSchemaLexer rikai.spark.sql.generated.RikaiModelSchemaParser rikai.spark.sql.generated.RikaiModelSchemaVisitor rikai.spark.sql.model rikai.spark.sql.schema rikai.spark.types rikai.spark.types.geometry rikai.spark.types.video rikai.spark.types.vision rikai.spark.utils rikai.tensorflow rikai.tensorflow.models rikai.testing rikai.testing.asserters rikai.testing.utils rikai.types rikai.types.geometry rikai.types.rle rikai.types.video rikai.types.vision rikai.viz module() (rikai.spark.types.Box2dType class method) (rikai.spark.types.Box3dType class method) (rikai.spark.types.geometry.Box2dType class method) (rikai.spark.types.geometry.Box3dType class method) (rikai.spark.types.geometry.MaskType class method) (rikai.spark.types.geometry.PointType class method) (rikai.spark.types.ImageType class method) (rikai.spark.types.MaskType class method) (rikai.spark.types.NDArrayType class method) (rikai.spark.types.PointType class method) (rikai.spark.types.SegmentType class method) (rikai.spark.types.video.VideoStreamType class method) (rikai.spark.types.video.YouTubeVideoType class method) (rikai.spark.types.VideoStreamType class method) (rikai.spark.types.vision.ImageType class method) (rikai.spark.types.YouTubeVideoType class method) move_tensor_to_device() (in module rikai.spark.sql.codegen.pytorch) N name (rikai.spark.sql.model.ModelSpec property) NDArrayType (class in rikai.spark.types) normalize_uri() (in module rikai.internal.uri_utils) numpy_to_image() (in module rikai.spark.functions.vision) O ObjectDetectionModelType (class in rikai.pytorch.models.torch) open() (rikai.mixin.Asset method) open_output_stream() (in module rikai.io) open_uri() (in module rikai.io) options (rikai.spark.sql.model.ModelSpec property) P PandasDataset (class in rikai.pytorch.pandas) parse_schema() (in module rikai.spark.sql.schema) parser (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.ArrayTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.FieldTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.IdentifierContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedArrayTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedStructTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.PlainFieldTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.SchemaContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructFieldContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructTypeContext attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.UnquotedIdentifierContext attribute) Point (class in rikai.types.geometry) PointType (class in rikai.spark.types) (class in rikai.spark.types.geometry) POLYGON (rikai.types.geometry.Mask.Type attribute) post_processing() (in module rikai.contrib.torch.transforms.fasterrcnn_resnet50_fpn) (in module rikai.contrib.torch.transforms.ssd) pre_processing() (in module rikai.contrib.torch.transforms.fasterrcnn_resnet50_fpn) (in module rikai.contrib.torch.transforms.ssd) predict() (rikai.contrib.tfhub.tensorflow.ssd.SSDModelType method) (rikai.contrib.torchhub.pytorch.vision.resnet.ResnetModelType method) (rikai.pytorch.models.feature_extractor.FeatureExtractorType method) (rikai.pytorch.models.ssd_class_scores.SSDClassScoresModelType method) (rikai.pytorch.models.torch.ClassificationModelType method) (rikai.pytorch.models.torch.ObjectDetectionModelType method) (rikai.sklearn.models.Classification method) (rikai.sklearn.models.DimensionalityReduction method) (rikai.sklearn.models.Regression method) (rikai.spark.sql.model.ModelType method) Pretrained (class in rikai.mixin) pretrained_model() (rikai.mixin.Pretrained method) (rikai.pytorch.models.torch.TorchModelType method) R read() (rikai.types.vision.Image static method) register() (in module rikai.parquet.resolver) (rikai.parquet.resolver.Resolver class method) Registry (class in rikai.spark.sql.codegen.base) Regression (class in rikai.sklearn.models) release() (rikai.pytorch.models.torch.TorchModelType method) (rikai.spark.sql.model.ModelType method) reset() (rikai.parquet.resolver.Resolver class method) ResnetModelType (class in rikai.contrib.torchhub.pytorch.vision.resnet) resolve() (rikai.parquet.resolver.BaseResolver method) (rikai.parquet.resolver.DefaultResolver method) (rikai.parquet.resolver.Resolver class method) (rikai.spark.sql.codegen.base.Registry method) Resolver (class in rikai.parquet.resolver) rikai module rikai.conf module rikai.contrib module rikai.contrib.datasets module rikai.contrib.datasets.coco module rikai.contrib.tfhub module rikai.contrib.tfhub.tensorflow module rikai.contrib.tfhub.tensorflow.ssd module rikai.contrib.torch module rikai.contrib.torch.detections module rikai.contrib.torch.transforms module rikai.contrib.torch.transforms.fasterrcnn_resnet50_fpn module rikai.contrib.torch.transforms.ssd module rikai.contrib.torchhub module rikai.contrib.torchhub.pytorch module rikai.contrib.torchhub.pytorch.vision module rikai.contrib.torchhub.pytorch.vision.resnet module rikai.contrib.video module rikai.contrib.video.functions module rikai.exceptions module rikai.internal module rikai.internal.reflection module rikai.internal.uri_utils module rikai.io module rikai.logging module rikai.mixin module rikai.numpy module rikai.parquet module rikai.parquet.dataset module rikai.parquet.resolver module rikai.parquet.writer module rikai.pytorch module rikai.pytorch.data module rikai.pytorch.models module rikai.pytorch.models.convnext module rikai.pytorch.models.efficientnet module rikai.pytorch.models.fasterrcnn module rikai.pytorch.models.feature_extractor module rikai.pytorch.models.keypointrcnn module rikai.pytorch.models.maskrcnn module rikai.pytorch.models.resnet module rikai.pytorch.models.retinanet module rikai.pytorch.models.ssd module rikai.pytorch.models.ssd_class_scores module rikai.pytorch.models.torch module rikai.pytorch.pandas module rikai.pytorch.transforms module rikai.pytorch.vision module rikai.sklearn module rikai.sklearn.models module rikai.spark module rikai.spark.functions module rikai.spark.functions.geometry module rikai.spark.functions.io module rikai.spark.functions.vision module rikai.spark.sql module rikai.spark.sql.codegen module rikai.spark.sql.codegen.base module rikai.spark.sql.codegen.dummy module rikai.spark.sql.codegen.fs module rikai.spark.sql.codegen.mlflow_logger module rikai.spark.sql.codegen.pytorch module rikai.spark.sql.codegen.sklearn module rikai.spark.sql.codegen.testing module rikai.spark.sql.exceptions module rikai.spark.sql.generated module rikai.spark.sql.generated.RikaiModelSchemaLexer module rikai.spark.sql.generated.RikaiModelSchemaParser module rikai.spark.sql.generated.RikaiModelSchemaVisitor module rikai.spark.sql.model module rikai.spark.sql.schema module rikai.spark.types module rikai.spark.types.geometry module rikai.spark.types.video module rikai.spark.types.vision module rikai.spark.utils module rikai.tensorflow module rikai.tensorflow.models module rikai.testing module rikai.testing.asserters module rikai.testing.utils module rikai.types module rikai.types.geometry module rikai.types.rle module rikai.types.video module rikai.types.vision module rikai.viz module RikaiModelSchemaLexer (class in rikai.spark.sql.generated.RikaiModelSchemaLexer) RikaiModelSchemaParser (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.ArrayContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.ArrayTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.FieldContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.FieldTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.IdentifierContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.NestedArrayTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.NestedStructTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.PlainFieldTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.SchemaContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.StructContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.StructFieldContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.StructTypeContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaParser.UnquotedIdentifierContext (class in rikai.spark.sql.generated.RikaiModelSchemaParser) RikaiModelSchemaVisitor (class in rikai.spark.sql.generated.RikaiModelSchemaVisitor) RikaiToTensor (class in rikai.pytorch.transforms) RLE (rikai.types.geometry.Mask.Type attribute) RULE_array (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) RULE_field (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) RULE_fieldType (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) RULE_identifier (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) RULE_schema (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) RULE_struct (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) ruleNames (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) S save() (rikai.types.vision.Image method) scalaUDT() (rikai.spark.types.Box2dType class method) (rikai.spark.types.Box3dType class method) (rikai.spark.types.geometry.Box2dType class method) (rikai.spark.types.geometry.Box3dType class method) (rikai.spark.types.geometry.MaskType class method) (rikai.spark.types.geometry.PointType class method) (rikai.spark.types.ImageType class method) (rikai.spark.types.MaskType class method) (rikai.spark.types.NDArrayType class method) (rikai.spark.types.PointType class method) (rikai.spark.types.SegmentType class method) (rikai.spark.types.video.VideoStreamType class method) (rikai.spark.types.video.YouTubeVideoType class method) (rikai.spark.types.VideoStreamType class method) (rikai.spark.types.vision.ImageType class method) (rikai.spark.types.YouTubeVideoType class method) scale() (rikai.types.vision.Image method) scene_detect() (in module rikai.contrib.video.functions) schema (rikai.spark.sql.model.ModelSpec property) schema() (rikai.contrib.tfhub.tensorflow.ssd.SSDModelType method) (rikai.contrib.torchhub.pytorch.vision.resnet.ResnetModelType method) (rikai.pytorch.models.feature_extractor.FeatureExtractorType method) (rikai.pytorch.models.ssd_class_scores.SSDClassScoresModelType method) (rikai.pytorch.models.torch.ClassificationModelType method) (rikai.pytorch.models.torch.ObjectDetectionModelType method) (rikai.sklearn.models.Classification method) (rikai.sklearn.models.DimensionalityReduction method) (rikai.sklearn.models.Regression method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) (rikai.spark.sql.model.ModelType method) SchemaError Segment (class in rikai.types.video) SegmentType (class in rikai.spark.types) serialize() (rikai.spark.types.Box2dType method) (rikai.spark.types.Box3dType method) (rikai.spark.types.geometry.Box2dType method) (rikai.spark.types.geometry.Box3dType method) (rikai.spark.types.geometry.MaskType method) (rikai.spark.types.geometry.PointType method) (rikai.spark.types.ImageType method) (rikai.spark.types.MaskType method) (rikai.spark.types.NDArrayType method) (rikai.spark.types.PointType method) (rikai.spark.types.SegmentType method) (rikai.spark.types.video.VideoStreamType method) (rikai.spark.types.video.YouTubeVideoType method) (rikai.spark.types.VideoStreamType method) (rikai.spark.types.vision.ImageType method) (rikai.spark.types.YouTubeVideoType method) serializedATN() (in module rikai.spark.sql.generated.RikaiModelSchemaLexer) (in module rikai.spark.sql.generated.RikaiModelSchemaParser) set_default_scheme() (rikai.parquet.resolver.Resolver class method) sharedContextCache (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) simpleString() (rikai.spark.types.Box2dType method) (rikai.spark.types.Box3dType method) (rikai.spark.types.geometry.Box2dType method) (rikai.spark.types.geometry.Box3dType method) (rikai.spark.types.geometry.MaskType method) (rikai.spark.types.geometry.PointType method) (rikai.spark.types.ImageType method) (rikai.spark.types.MaskType method) (rikai.spark.types.NDArrayType method) (rikai.spark.types.PointType method) (rikai.spark.types.SegmentType method) (rikai.spark.types.video.VideoStreamType method) (rikai.spark.types.video.YouTubeVideoType method) (rikai.spark.types.VideoStreamType method) (rikai.spark.types.vision.ImageType method) (rikai.spark.types.YouTubeVideoType method) SingleFrameSampler (class in rikai.types.video) SklearnModelType (class in rikai.sklearn.models) SPARK_PARQUET_ROW_METADATA (rikai.parquet.dataset.Dataset attribute) (rikai.parquet.resolver.DefaultResolver attribute) SparkDataTypeVisitor (class in rikai.spark.sql.schema) SpecError spectrogram_image() (in module rikai.spark.functions.vision) sqlType() (rikai.spark.types.Box2dType class method) (rikai.spark.types.Box3dType class method) (rikai.spark.types.geometry.Box2dType class method) (rikai.spark.types.geometry.Box3dType class method) (rikai.spark.types.geometry.MaskType class method) (rikai.spark.types.geometry.PointType class method) (rikai.spark.types.ImageType class method) (rikai.spark.types.MaskType class method) (rikai.spark.types.NDArrayType class method) (rikai.spark.types.PointType class method) (rikai.spark.types.SegmentType class method) (rikai.spark.types.video.VideoStreamType class method) (rikai.spark.types.video.YouTubeVideoType class method) (rikai.spark.types.VideoStreamType class method) (rikai.spark.types.vision.ImageType class method) (rikai.spark.types.YouTubeVideoType class method) SSDClassScoresExtractor (class in rikai.pytorch.models.ssd_class_scores) SSDClassScoresModelType (class in rikai.pytorch.models.ssd_class_scores) SSDModelType (class in rikai.contrib.tfhub.tensorflow.ssd) STRUCT (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) struct() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser method) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.NestedStructTypeContext method) STRUCT() (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser.StructTypeContext method) Style (class in rikai.viz) symbolicNames (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) T T__0 (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) T__1 (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) T__2 (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) T__3 (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) TensorflowModelType (class in rikai.tensorflow.models) TestModel (class in rikai.spark.sql.codegen.testing) Text (class in rikai.viz) to_dict() (rikai.mixin.ToDict method) (rikai.types.geometry.Box2d method) (rikai.types.geometry.Box3d method) (rikai.types.geometry.Mask method) (rikai.types.geometry.Point method) (rikai.types.video.VideoStream method) (rikai.types.vision.Image method) to_embedded() (rikai.types.vision.Image method) to_image() (in module rikai.spark.functions.vision) to_mask() (rikai.types.geometry.Mask method) to_numpy() (rikai.mixin.ToNumpy method) (rikai.types.geometry.Box2d method) (rikai.types.geometry.Box3d method) (rikai.types.geometry.Mask method) (rikai.types.geometry.Point method) (rikai.types.vision.Image method) to_pandas() (rikai.parquet.dataset.Dataset method) to_pil() (rikai.types.vision.Image method) ToDict (class in rikai.mixin) ToNumpy (class in rikai.mixin) TorchModelType (class in rikai.pytorch.models.torch) training (rikai.pytorch.models.feature_extractor.FeatureExtractor attribute) (rikai.pytorch.models.ssd_class_scores.SSDClassScoresExtractor attribute) transform() (rikai.contrib.tfhub.tensorflow.ssd.SSDModelType method) (rikai.contrib.torchhub.pytorch.vision.resnet.ResnetModelType method) (rikai.pytorch.models.feature_extractor.FeatureExtractorType method) (rikai.pytorch.models.torch.ClassificationModelType method) (rikai.pytorch.models.torch.ObjectDetectionModelType method) (rikai.sklearn.models.SklearnModelType method) (rikai.spark.sql.model.ModelType method) U uri (rikai.mixin.Asset attribute) uri_equal() (in module rikai.internal.uri_utils) V validate() (rikai.spark.sql.codegen.dummy.DummyModelSpec method) (rikai.spark.sql.model.ModelSpec method) version (rikai.spark.sql.model.ModelSpec property) video_metadata() (in module rikai.spark.functions.vision) video_to_images() (in module rikai.spark.functions.vision) VideoSampler (class in rikai.types.video) VideoStream (class in rikai.types.video) VideoStreamType (class in rikai.spark.types) (class in rikai.spark.types.video) view() (in module rikai.numpy) visitArrayType() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) (rikai.spark.sql.schema.SparkDataTypeVisitor method) visitNestedArrayType() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) visitNestedStructType() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) visitPlainFieldType() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) (rikai.spark.sql.schema.SparkDataTypeVisitor method) visitSchema() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) visitStructField() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) (rikai.spark.sql.schema.SparkDataTypeVisitor method) visitStructType() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) (rikai.spark.sql.schema.SparkDataTypeVisitor method) visitUnquotedIdentifier() (rikai.spark.sql.generated.RikaiModelSchemaVisitor.RikaiModelSchemaVisitor method) (rikai.spark.sql.schema.SparkDataTypeVisitor method) W width (rikai.types.geometry.Box2d property) (rikai.types.geometry.Box3d attribute) with_label() (rikai.types.geometry.Box2d method) WS (rikai.spark.sql.generated.RikaiModelSchemaLexer.RikaiModelSchemaLexer attribute) (rikai.spark.sql.generated.RikaiModelSchemaParser.RikaiModelSchemaParser attribute) X x (rikai.types.geometry.Point attribute) xmax (rikai.types.geometry.Box2d attribute) xmin (rikai.types.geometry.Box2d attribute) Y y (rikai.types.geometry.Point attribute) ymax (rikai.types.geometry.Box2d attribute) ymin (rikai.types.geometry.Box2d attribute) YouTubeVideo (class in rikai.types.video) YouTubeVideoType (class in rikai.spark.types) (class in rikai.spark.types.video) Z z (rikai.types.geometry.Point attribute)