net

Documentation

Index

• type Net
  • func New(text string) (*Net, error)
  • func (net *Net) CopyTrainedLayersFromFile(file string) error
  • func (net *Net) CopyTrainedLayersFromParam(param *pb.NetParameter) error
  • func (net *Net) Forward(bottom []*blob.Blob) ([]*blob.Blob, error)
  • func (net *Net) ForwardFromTo(bottom []*blob.Blob, end int) (top []*blob.Blob, err error)
  • func (net *Net) GetInputSize() (int64, int64)
  • func (net *Net) Name() string

Types

type Net

type Net struct {
	Parameters *pb.NetParameter
	// contains filtered or unexported fields
}

func New

func New(text string) (*Net, error)

func (*Net) CopyTrainedLayersFromFile

func (net *Net) CopyTrainedLayersFromFile(file string) error

func (*Net) CopyTrainedLayersFromParam

func (net *Net) CopyTrainedLayersFromParam(param *pb.NetParameter) error

func (*Net) Forward

func (net *Net) Forward(bottom []*blob.Blob) ([]*blob.Blob, error)

func (*Net) ForwardFromTo

func (net *Net) ForwardFromTo(bottom []*blob.Blob, end int) (top []*blob.Blob, err error)

func (*Net) GetInputSize

func (net *Net) GetInputSize() (int64, int64)

func (*Net) Name

func (net *Net) Name() string