README
¶
go-caffe2 
This provides caffe2 bindings for Go.
You need to install caffe2 to your system first.
$ go get -u github.com/d4l3k/go-caffe2
See TestNet for an example of creating a new net with blobs and running it: https://github.com/d4l3k/go-caffe2/blob/master/net_test.go
Development
Check the code out with submodules
$ git clone --recurse-submodules -j16 git@github.com:d4l3k/go-caffe2.git
Install go protoc: https://github.com/golang/protobuf#installation
Generate protobuf sources and test:
make
License
go-caffe2 is licensed under the MIT license.
Documentation
¶
Index ¶
- type Blob
- type Net
- type Workspace
- func (w *Workspace) CreateBlob(blob string) *Blob
- func (w *Workspace) CreateLocalBlob(blob string) *Blob
- func (w *Workspace) CreateNet(net *caffe2pb.NetDef) (*Net, error)
- func (w *Workspace) DeleteNet(name string)
- func (w *Workspace) GetBlob(blob string) *Blob
- func (w *Workspace) GetNet(name string) *Net
- func (w *Workspace) HasBlob(blob string) bool
- func (p *Workspace) NewChild() *Workspace
- func (w *Workspace) RemoveBlob(blob string) error
- func (w *Workspace) RunNet(name string) error
- func (w *Workspace) RunPlan(plan *caffe2pb.PlanDef) error
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Blob ¶
type Blob struct {
// contains filtered or unexported fields
}
func (*Blob) Deserialize ¶
type Workspace ¶
type Workspace struct {
// contains filtered or unexported fields
}
func NewWorkspace ¶
func NewWorkspace() *Workspace
NewWorkspace creates a new workspace. The underlying C++ workspace is freed via a finalizer and all child Blobs and Nets have a pointer to the workspace so there won't ever be any unsafe memory accesses.
func (*Workspace) CreateBlob ¶
func (*Workspace) CreateLocalBlob ¶
func (*Workspace) NewChild ¶
NewChild returns a new child workspace that inherits blobs from the parent workspace.
Source Files
Directories