|
@@ -45,6 +45,12 @@ build(){
|
|
|
;;
|
|
|
labelImage )
|
|
|
git clone -b v2.2.0-rc3 --depth 1 https://github.com/tensorflow/tensorflow.git /tmp/tensorflow;
|
|
|
+ if [ "$(uname -m)" = "x86_64" ]; then
|
|
|
+ mv $(pwd)/extensions/functions/labelImage/dependencies/amd64/*.so $(pwd)/extensions/functions/labelImage/lib
|
|
|
+ fi;
|
|
|
+ if [ "$(uname -m)" = "aarch64" ]; then
|
|
|
+ mv $(pwd)/extensions/functions/labelImage/dependencies/arm64/*.so $(pwd)/extensions/functions/labelImage/lib
|
|
|
+ fi;
|
|
|
CGO_CFLAGS=-I/tmp/tensorflow CGO_LDFLAGS=-L$(pwd)/extensions/functions/labelImage/lib go build -trimpath -modfile extensions.mod --buildmode=plugin -o extensions/functions/labelImage/labelImage.so extensions/functions/labelImage/*.go
|
|
|
;;
|
|
|
* )
|