Send base64 to DeepDetect

Author: riptl

README.md

@@ -14,14 +14,7 @@ Because sometimes, you have folders full of badly named pictures, and you want t
 You need DeepDetect installed, the easiest way is using docker:
 ```
 docker pull beniz/deepdetect_cpu
-docker run -d -p 8080:8080 -v /path/to/images:/path/to/images beniz/deepdetect_cpu
-```
-
-PLEASE NOTE THAT THE PATH IN THE HOST SHOULD BE THE SAME IN THE CONTAINER!
-
-Example:
-```
-docker run -d -p 8080:8080 -v /home/corentin/Images:/home/corentin/Images beniz/deepdetect_cpu
+docker run -d -p 8080:8080 beniz/deepdetect_cpu
 ```
 
 Right now, the only supported installation of DeepDetect that works with DeepSort is the deepdetect_cpu container.

cmd/deepsort/classify.go

@@ -15,8 +15,8 @@ import (
 func googleNetClassification(path string, arguments *Arguments, wg *sync.WaitGroup) {
 	defer wg.Done()
 	url := arguments.URL + "/predict"
-	path, _ = filepath.Abs(path)
-	var jsonStr = []byte(`{"service":"deepsort-resnet","parameters":{"input":{"width":224,"height":224},"output":{"best":1},"mllib":{"gpu":false}},"data":["` + path + `"]}`)
+	dataStr := readFile(path)
+	var jsonStr = []byte(`{"service":"deepsort-resnet","parameters":{"input":{"width":224,"height":224},"output":{"best":1},"mllib":{"gpu":false}},"data":["` + dataStr + `"]}`)
 	// DEBUG
 	//fmt.Println("Request: " + string(jsonStr))
 	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonStr))
@@ -51,8 +51,8 @@ func googleNetClassification(path string, arguments *Arguments, wg *sync.WaitGro
 func resNet50Classification(path string, arguments *Arguments, wg *sync.WaitGroup) {
 	defer wg.Done()
 	url := arguments.URL + "/predict"
-	path, _ = filepath.Abs(path)
-	var jsonStr = []byte(`{"service":"deepsort-resnet-50","parameters":{"input":{"width":224,"height":224},"output":{"best":1},"mllib":{"gpu":false}},"data":["` + path + `"]}`)
+	dataStr := readFile(path)
+	var jsonStr = []byte(`{"service":"deepsort-resnet-50","parameters":{"input":{"width":224,"height":224},"output":{"best":1},"mllib":{"gpu":false}},"data":["` + dataStr + `"]}`)
 	// DEBUG
 	//fmt.Println("Request: " + string(jsonStr))
 	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonStr))

cmd/deepsort/read_file.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"bytes"
+	"encoding/base64"
+	"github.com/CorentinB/DeepSort/pkg/logging"
+	"io"
+	"os"
+	"path/filepath"
+)
+
+// Reads file and returns data string for DeepDetect
+func readFile(filePath string) string {
+	f, err := os.Open(filePath)
+	if err != nil {
+		logging.Error("Can't open the file.", "["+filepath.Base(filePath)+"]")
+		os.Exit(1)
+	}
+	defer f.Close()
+
+	var buf bytes.Buffer
+	enc := base64.NewEncoder(base64.StdEncoding, &buf)
+
+	if _, err := io.Copy(enc, f); err != nil {
+		logging.Error("Can't read the file.", "["+filepath.Base(filePath)+"]")
+		os.Exit(1)
+	}
+
+	enc.Close()
+	return buf.String()
+}