gopt

README

GoPt

Overview

GoPt is a small library for loading PyTorch image classification models into Golang code.

Dependencies

• Libtorch C++ v1.11.0 library of Pytorch

Installation

• Default CUDA version is 11.3 if CUDA is available otherwise using CPU version.
• Default Pytorch C++ API version is 1.11.0

NOTE: libtorch will be installed at /usr/local/lib

CPU

Step 1: Setup libtorch (skip this step if a valid libtorch already installed in your machine!)

    wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-libtorch.sh
    chmod +x setup-libtorch.sh
    export CUDA_VER=cpu && bash setup-libtorch.sh

Update Environment: in Debian/Ubuntu, add/update the following lines to .bashrc file

    export GOTCH_LIBTORCH="/usr/local/lib/libtorch"
    export LIBRARY_PATH="$LIBRARY_PATH:$GOTCH_LIBTORCH/lib"
    export CPATH="$CPATH:$GOTCH_LIBTORCH/lib:$GOTCH_LIBTORCH/include:$GOTCH_LIBTORCH/include/torch/csrc/api/include"
    export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$GOTCH_LIBTORCH/lib"

Step 2: Setup gotch

    wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-gotch.sh
    chmod +x setup-gotch.sh
    export CUDA_VER=cpu && export GOTCH_VER=v0.7.0 && bash setup-gotch.sh

Example

package main

import (
	"flag"
	"fmt"

	"github.com/jbloxsome/gopt"
)

var (
	modelPath string
	imageFile string
)

func init() {
	flag.StringVar(&modelPath, "modelpath", "./model.pt", "full path to exported pytorch model.")
	flag.StringVar(&imageFile, "image", "./image.jpg", "full path to image file.")
}

func main() {
	flag.Parse()

	labels := []string{
		"false",
		"true",
	}

	model, err := gopt.NewGoPt(modelPath, labels)
	if err != nil {
		log.Fatal(err)
	}

	pred, err := model.Predict(imageFile)
	if err != nil {
		fmt.Println(err)
	}

	fmt.Println(pred)
}

Documentation

Index

• func GetFileContentType(path string) (string, error)
• type GoPt
  • func NewGoPt(modelPath string, labels []string) (*GoPt, error)
  • func (gopt *GoPt) Predict(path string) (string, error)

Functions

func GetFileContentType

func GetFileContentType(path string) (string, error)

Types

type GoPt

type GoPt struct {
	Model  *ts.CModule
	Labels []string
	Imnet  *vision.ImageNet
}

func NewGoPt

func NewGoPt(modelPath string, labels []string) (*GoPt, error)

func (*GoPt) Predict

func (gopt *GoPt) Predict(path string) (string, error)