apps/go-darknet
1
0
Fork 0
mirror of https://github.com/LdDl/go-darknet.git synced 2025-10-06 16:17:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

free_network_ptr instead of free_network

Browse Source
This commit is contained in:
Dimitrii
2021-01-30 13:57:21 +03:00
parent 13a0697585
commit 40df8cce91
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/base_example/main.go
View File

@@ -91,6 +91,8 @@ func main() {
// } // }
} }
} }
n.Close()
} }
func imageToBytes(img image.Image) ([]byte, error) { func imageToBytes(img image.Image) ([]byte, error) {

4
network.go
View File

@@ -58,7 +58,7 @@ func (n *YOLONetwork) Close() error {
if n.cNet == nil { if n.cNet == nil {
return errNetworkNotInit return errNetworkNotInit
} }
C.free_network(*n.cNet) C.free_network_ptr(n.cNet)
n.cNet = nil n.cNet = nil
return nil return nil
} }
@@ -80,4 +80,4 @@ func (n *YOLONetwork) Detect(img *DarknetImage) (*DetectionResult, error) {
OverallTimeTaken: endTimeOverall.Sub(startTime), OverallTimeTaken: endTimeOverall.Sub(startTime),
} }
return &out, nil return &out, nil
} }