mirror of
https://github.com/LdDl/go-darknet.git
synced 2025-09-27 03:56:18 +08:00
29 lines
549 B
Go
29 lines
549 B
Go
package darknet
|
|
|
|
// #include <stdlib.h>
|
|
// #include "classnames.h"
|
|
import "C"
|
|
import "unsafe"
|
|
|
|
func freeClassNames(names **C.char) {
|
|
C.free_class_names(names)
|
|
}
|
|
|
|
func loadClassNames(dataConfigFile string) **C.char {
|
|
d := C.CString(dataConfigFile)
|
|
defer C.free(unsafe.Pointer(d))
|
|
|
|
return C.read_class_names(d)
|
|
}
|
|
|
|
func makeClassNames(names **C.char, classes int) []string {
|
|
out := make([]string, classes)
|
|
for i := 0; i < classes; i++ {
|
|
n := C.get_class_name(names, C.int(i), C.int(classes))
|
|
s := C.GoString(n)
|
|
out[i] = s
|
|
}
|
|
|
|
return out
|
|
}
|