| | |
| | | flag int |
| | | |
| | | handle interface{} |
| | | fnInit func(int, int, int) interface{} |
| | | fnInit func(int, int, int, func(...interface{})) interface{} |
| | | fnFree func(interface{}) |
| | | fnRun func(interface{}, []byte, int, int, int) []sdkstruct.FgResult |
| | | fnProcess func(interface{}, []sdkstruct.SDKImage) ([]sdkstruct.FgResult, error) |
| | |
| | | flag: flag, |
| | | |
| | | handle: nil, |
| | | fnInit: fnInit.(func(int, int, int) interface{}), |
| | | fnInit: fnInit.(func(int, int, int, func(...interface{})) interface{}), |
| | | fnFree: fnFree.(func(interface{})), |
| | | fnRun: fnRun.(func(interface{}, []byte, int, int, int) []sdkstruct.FgResult), |
| | | fnProcess: fnProc.(func(interface{}, []sdkstruct.SDKImage) ([]sdkstruct.FgResult, error)), |
| | |
| | | if gpu == -1 { |
| | | gpu = gogpu.ValidGPU(2048) |
| | | } |
| | | h := t.fnInit(gpu, t.batchSize, t.flag) |
| | | h := t.fnInit(gpu, t.batchSize, t.flag, logo.Infoln) |
| | | logo.Infoln("HumanTrack USE GPU: ", gpu) |
| | | |
| | | if h == nil { |