取消张量生成

README.md

@@ -1,3 +1,48 @@
 # webp
 
 提供webp生成服务
+
+```bash
+# 需要安装依赖
+sudo apt install libopencv-dev
+```
+
+防止错误 ‘ArucoDetector’ in namespace ‘cv::aruco’ does not name a type
+將安裝的 opencv 版本從 4.2 更改為 4.7，它工作正常。
+
+```bash
+# E: 无法定位软件包 libdc1394-22-dev
+sudo add-apt-repository "deb http://security.ubuntu.com/ubuntu xenial-security main"
+sudo apt update
+sudo apt upgrade
+sudo apt install libjasper1 libjasper-dev
+
+# 添加主服务器
+sudo gedit /etc/apt/sources.list
+    deb http://archive.ubuntu.com/ubuntu/ trusty main universe restricted multiverse
+
+sudo apt-get update
+sudo apt update
+sudo apt upgrade
+
+# GPT 方案
+sudo add-apt-repository universe
+sudo apt update
+sudo apt install libdc1394-22-dev
+apt search libdc1394
+
+# 手动下载 https://pkgs.org/download/libdc1394-22
+wget http://archive.ubuntu.com/ubuntu/pool/universe/libd/libdc1394-22/libdc1394-22_2.2.5-2.1_amd64.deb
+sudo dpkg -i libdc1394-22_2.2.5-2.1_amd64.deb
+apt search libdc1394
+
+# 确认安装后注释 Markfile 第52行
+cd gocv
+make install
+
+# 下载超时请爬墙
+```
+
+torch 模型文件
+https://d2j0dndfm35trm.cloudfront.net/resnet-50.t7
+https://github.com/facebookarchive/fb.resnet.torch/tree/master/pretrained

go.mod

@@ -31,5 +31,7 @@ require (
 	github.com/chai2010/webp v1.1.1
 	github.com/go-sql-driver/mysql v1.7.0
 	github.com/sizeofint/gif-to-webp v0.0.0-20210224202734-e9d7ed071591
+	github.com/wangkuiyi/gotorch v0.0.0-20201028015551-9afed2f3ad7b
+	gocv.io/x/gocv v0.32.1
 	golang.org/x/image v0.0.0-20211028202545-6944b10bf410 // indirect
 )

go.sum

@@ -128,6 +128,7 @@ github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ
 github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
 github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
 github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
+github.com/hybridgroup/mjpeg v0.0.0-20140228234708-4680f319790e/go.mod h1:eagM805MRKrioHYuU7iKLUyFPVKqVV6um5DAvCkUtXs=
 github.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=
 github.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=
 github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU=
@@ -141,6 +142,7 @@ github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0V
 github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
 github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
 github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
+github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=
 github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU=
 github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
@@ -168,12 +170,16 @@ github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpE
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
+github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 github.com/subosito/gotenv v1.4.2 h1:X1TuBLAMDFbaTAChgCBLu3DU3UPyELpnF2jjJ2cz/S8=
 github.com/subosito/gotenv v1.4.2/go.mod h1:ayKnFf/c6rvx/2iiLrJUk1e6plDbT3edrFNGqEflhK0=
+github.com/wangkuiyi/gotorch v0.0.0-20201028015551-9afed2f3ad7b h1:oJfm5gCGdy9k2Yb+qmMR+HMRQ89CbVDsDi6DD9AZSTk=
+github.com/wangkuiyi/gotorch v0.0.0-20201028015551-9afed2f3ad7b/go.mod h1:WC7g+ojb7tPOZhHI2+ZI7ZXTW7uzF9uFOZfZgIX+SjI=
+github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
 github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
@@ -184,6 +190,9 @@ go.opencensus.io v0.22.2/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=
 go.opencensus.io v0.22.3/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=
 go.opencensus.io v0.22.4/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=
 go.opencensus.io v0.22.5/go.mod h1:5pWMHQbX5EPX2/62yrJeAkowc+lfs/XD7Uxpq3pI6kk=
+gocv.io/x/gocv v0.24.0/go.mod h1:Rar2PS6DV+T4FL+PM535EImD/h13hGVaHhnCu1xarBs=
+gocv.io/x/gocv v0.32.1 h1:BC9hHs5+47nVgySUFVKntc6RsF3SULFzqk6OV9xz+C0=
+gocv.io/x/gocv v0.32.1/go.mod h1:oc6FvfYqfBp99p+yOEzs9tbYF9gOrAQSeL/dyIPefJU=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
 golang.org/x/crypto v0.0.0-20190605123033-f99c8df09eb5/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=

models/resnet.go

@@ -0,0 +1,55 @@
+package models
+
+import (
+	//"gocv.io/x/gocv"
+	//"github.com/xuyu/gotool/torch"
+	"fmt"
+
+	"github.com/wangkuiyi/gotorch"
+)
+
+func init() {
+	// 模型文件地址: https://download.pytorch.org/models/resnet50-19c8e357.pth
+	// 模型地址: "/home/satori/webp/data/resnet-50.t7"
+
+	tensor := gotorch.Load("/home/satori/webp/data/resnet-50.t7")
+	fmt.Println(tensor)
+
+	//model := torch.NewModel()
+	//err := model.ReadFromFile("/home/satori/webp/data/resnet-50.t7")
+	//if err != nil {
+	//	panic(err)
+	//}
+
+	/**
+	t7 := "/home/satori/webp/data/resnet-50.t7"
+	// 加载t7格式的模型
+	model := gocv.ReadNetFromTorch(t7)
+
+	if model.Empty() {
+		panic("Failed to load model")
+	}
+
+	fmt.Println("==============================")
+	img := gocv.IMRead("data/test.jpeg", gocv.IMReadColor)
+	if img.Empty() {
+		panic("Failed to read image")
+	}
+
+	fmt.Println("==============================")
+	inputBlob := gocv.BlobFromImage(img, 1.0, image.Pt(224, 224), gocv.NewScalar(0, 0, 0, 0), true, false)
+	defer inputBlob.Close()
+
+	fmt.Println("==============================")
+	model.SetInput(inputBlob, "input")
+	outputBlob := model.Forward("output")
+	defer outputBlob.Close()
+
+	fmt.Println("==============================")
+	features := outputBlob.Reshape(1, 1)
+	defer features.Close()
+
+	fmt.Println("==============================")
+	fmt.Println(features.ToBytes())
+	**/
+}