fix build

mattn · Oct 16, 2023 · edc3191 · edc3191
1 parent 44cbb69
commit edc3191
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 10 deletions.
diff --git a/_example/ssd/main.go b/_example/ssd/main.go
@@ -102,9 +102,11 @@ func detect(ctx context.Context, wg *sync.WaitGroup, resultChan chan<- *ssdResul
 			}
 		} else {
 			gocv.Resize(frame, &resized, image.Pt(wanted_width, wanted_height), 0, 0, gocv.InterpolationDefault)
-			if v, err := resized.DataPtrUint8(); err == nil {
-				copy(input.UInt8s(), v)
-			}
+			//if v, err := resized.DataPtrUint8(); err == nil {
+			//	copy(input.UInt8s(), v)
+			//}
+			v := resized.DataPtrUint8()
+			copy(input.UInt8s(), v)
 		}
 		resized.Close()
 		status := interpreter.Invoke()

diff --git a/_example/ssd_edgetpu/main.go b/_example/ssd_edgetpu/main.go
@@ -104,9 +104,11 @@ func detect(ctx context.Context, wg *sync.WaitGroup, resultChan chan<- *ssdResul
 			}
 		} else {
 			gocv.Resize(frame, &resized, image.Pt(wanted_width, wanted_height), 0, 0, gocv.InterpolationDefault)
-			if v, err := resized.DataPtrUint8(); err == nil {
-				copy(input.UInt8s(), v)
-			}
+			//if v, err := resized.DataPtrUint8(); err == nil {
+			//	copy(input.UInt8s(), v)
+			//}
+			v := resized.DataPtrUint8()
+			copy(input.UInt8s(), v)
 		}
 		resized.Close()
 		status := interpreter.Invoke()

diff --git a/_example/ssd_xnnpack/main.go b/_example/ssd_xnnpack/main.go
@@ -103,9 +103,10 @@ func detect(ctx context.Context, wg *sync.WaitGroup, resultChan chan<- *ssdResul
 			}
 		} else {
 			gocv.Resize(frame, &resized, image.Pt(wanted_width, wanted_height), 0, 0, gocv.InterpolationDefault)
-			if v, err := resized.DataPtrUint8(); err == nil {
-				copy(input.UInt8s(), v)
-			}
+			v := resized.DataPtrUint8()
+			copy(input.UInt8s(), v)
+			//v := resized.DataPtrUint8()
+			//copy(input.UInt8s(), v)
 		}
 		resized.Close()
 		status := interpreter.Invoke()
@@ -158,7 +159,7 @@ func main() {
 
 	options := tflite.NewInterpreterOptions()
 	options.AddDelegate(xnnpack.New(xnnpack.DelegateOptions{NumThreads: 4}))
-	//options.SetNumThread(4)
+	options.SetNumThread(8)
 	defer options.Delete()
 
 	interpreter := tflite.NewInterpreter(model, options)