| | |
| | | p := C.c_human_tracker_create(C.int(gpu), C.int(batchSize), C.int(flag)) |
| | | if p != nil { |
| | | res := C.init_fgres(C.int(batchSize)) |
| | | return &HumanTracker{p, res, batchSize} |
| | | return &HumanTracker{p, res, batchSize, nil} |
| | | } |
| | | return nil |
| | | } |
| | |
| | | logo.Infoln("batch~~~~~~", k, " image: ", v) |
| | | if v == nil { |
| | | if h.fakeImage == nil { |
| | | h.fakeImage = C.create_fake_image(wid * hei * chn) |
| | | h.fakeImage = C.create_fake_image(C.int(wid * hei * chn)) |
| | | } |
| | | C.fill_images(cImgs, C.int(h.batchSize), C.int(k), h.fakeImage, C.int(wid), C.int(hei), C.int(chn)) |
| | | } else { |