From b2640e6b3496a0f8fd4ce36548b8f5d8b74448ba Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期六, 11 一月 2020 15:38:51 +0800
Subject: [PATCH] update
---
goreid.go | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/goreid.go b/goreid.go
index 128f7b1..2784da5 100644
--- a/goreid.go
+++ b/goreid.go
@@ -1,11 +1,11 @@
-package goreid
+package main
/*
-#cgo CFLAGS: -I${SRCDIR}/sdk/include -I${SRCDIR}/opencv-4.0.1/include/opencv4 -I${SRCDIR}/sdk/include/torch/csrc/api/include -I${SRCDIR}/csrc -w -g
-#cgo CXXFLAGS: -I${SRCDIR}/sdk/include -I${SRCDIR}/opencv-4.0.1/include/opencv4 -I${SRCDIR}/sdk/include/torch/csrc/api/include -I${SRCDIR}/csrc -w -g -std=c++11
+#cgo CFLAGS: -I${SRCDIR}/sdk/include -I${SRCDIR}/thirdparty/include/opencv4 -I${SRCDIR}/sdk/include/torch/csrc/api/include -I${SRCDIR}/csrc -w -g
+#cgo CXXFLAGS: -I${SRCDIR}/sdk/include -I${SRCDIR}/thirdparty/include/opencv4 -I${SRCDIR}/sdk/include/torch/csrc/api/include -I${SRCDIR}/csrc -w -g -std=c++11
#cgo LDFLAGS: -Wl,-rpath,${SRCDIR}/sdk/lib
#cgo LDFLAGS: -L${SRCDIR}/sdk/lib -ltorch -lc10 -lc10_cuda -lcudart -lgomp -lnvToolsExt
-#cgo LDFLAGS: -L${SRCDIR}/opencv-4.0.1/lib -lopencv_core -lopencv_imgproc
+#cgo LDFLAGS: -L${SRCDIR}/thirdparty/lib -lopencv_core -lopencv_imgproc
#cgo LDFLAGS: -ldl -lpthread
#include <stdlib.h>
@@ -23,8 +23,8 @@
handle unsafe.Pointer
}
-// NewReID new reid
-func NewReID(gpu int, module string) *ReID {
+// NewSDK new reid
+func NewSDK(gpu int, module string) *ReID {
cmodule := C.CString(module)
defer C.free(unsafe.Pointer(cmodule))
h := C.create_reid(C.int(gpu), cmodule)
--
Gitblit v1.8.0