vision

Documentation

Overview

Package vision implements computer vision algorithms.

Index

• type ImageTrainingStore
  • func NewImageTrainingStore(ctx context.Context, mongoURI, db, collection string) (*ImageTrainingStore, error)
  • func (its *ImageTrainingStore) BuildIndexes(ctx context.Context) error
  • func (its *ImageTrainingStore) Close(ctx context.Context) error
  • func (its *ImageTrainingStore) GetImage(ctx context.Context, id primitive.ObjectID) (TrainingImage, error)
  • func (its *ImageTrainingStore) GetImagesForLabel(ctx context.Context, label string) ([]primitive.ObjectID, error)
  • func (its *ImageTrainingStore) GetLabels(ctx context.Context) (map[string]int, error)
  • func (its *ImageTrainingStore) SetLabelsForImage(ctx context.Context, id primitive.ObjectID, labels []string) error
  • func (its *ImageTrainingStore) StoreImage(ctx context.Context, img image.Image, metaData map[string]interface{}, ...) (primitive.ObjectID, error)
  • func (its *ImageTrainingStore) StoreImageFromDisk(ctx context.Context, fn string, labels []string) (primitive.ObjectID, error)
• type Object
  • func NewEmptyObject() *Object
  • func NewObject(cloud pc.PointCloud) (*Object, error)
  • func NewObjectWithLabel(cloud pc.PointCloud, label string, geometry *commonpb.Geometry) (*Object, error)
  • func (o *Object) Distance() (float64, error)
• type TrainingImage

Types

type ImageTrainingStore

type ImageTrainingStore struct {
	// contains filtered or unexported fields
}

ImageTrainingStore TODO.

func NewImageTrainingStore

func NewImageTrainingStore(ctx context.Context, mongoURI, db, collection string) (*ImageTrainingStore, error)

NewImageTrainingStore TODO.

func (*ImageTrainingStore) BuildIndexes

func (its *ImageTrainingStore) BuildIndexes(ctx context.Context) error

BuildIndexes TODO.

func (*ImageTrainingStore) Close

func (its *ImageTrainingStore) Close(ctx context.Context) error

Close TODO.

func (*ImageTrainingStore) GetImage

func (its *ImageTrainingStore) GetImage(ctx context.Context, id primitive.ObjectID) (TrainingImage, error)

GetImage TODO.

func (*ImageTrainingStore) GetImagesForLabel

func (its *ImageTrainingStore) GetImagesForLabel(ctx context.Context, label string) ([]primitive.ObjectID, error)

GetImagesForLabel TODO.

func (*ImageTrainingStore) GetLabels

func (its *ImageTrainingStore) GetLabels(ctx context.Context) (map[string]int, error)

GetLabels TODO.

func (*ImageTrainingStore) SetLabelsForImage

func (its *ImageTrainingStore) SetLabelsForImage(ctx context.Context, id primitive.ObjectID, labels []string) error

SetLabelsForImage TODO.

func (*ImageTrainingStore) StoreImage

func (its *ImageTrainingStore) StoreImage(
	ctx context.Context,
	img image.Image,
	metaData map[string]interface{},
	labels []string,
) (primitive.ObjectID, error)

StoreImage TODO.

func (*ImageTrainingStore) StoreImageFromDisk

func (its *ImageTrainingStore) StoreImageFromDisk(ctx context.Context, fn string, labels []string) (primitive.ObjectID, error)

StoreImageFromDisk TODO.

type Object

type Object struct {
	pc.PointCloud
	Geometry spatialmath.Geometry
}

Object extends PointCloud with respective metadata, like the center coordinate. NOTE(bh):Can potentially add category or pose information to this struct.

func NewEmptyObject

func NewEmptyObject() *Object

NewEmptyObject creates a new empty point cloud with metadata.

func NewObject

func NewObject(cloud pc.PointCloud) (*Object, error)

NewObject creates a new vision.Object from a point cloud with an empty label.

func NewObjectWithLabel

func NewObjectWithLabel(cloud pc.PointCloud, label string, geometry *commonpb.Geometry) (*Object, error)

NewObjectWithLabel creates a new vision.Object from a point cloud with the given label.

func (*Object) Distance

func (o *Object) Distance() (float64, error)

Distance calculates and returns the distance from the center point of the object to the origin.

type TrainingImage

type TrainingImage struct {
	ID       primitive.ObjectID `bson:"_id" json:"id,omitempty"`
	Data     []byte
	Labels   []string
	MetaData map[string]interface{}
}

TrainingImage TODO.