zhangmeng
2019-12-19 0ea7a18ff3cc53ae711ed2d142a1c755f8f2f959
update
1个文件已修改
4 ■■■ 已修改文件
gohumantrack/gohumantrack.go 4 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
gohumantrack/gohumantrack.go
@@ -25,6 +25,7 @@
void *create_fake_image(const int size){
    unsigned char *img = (unsigned char*)malloc(size);
    printf("batch~~~~~~ fake image len: %d\n", size);
    memset(img, 0, size);
    return img;
}
@@ -192,12 +193,13 @@
    var freeFakeImage []unsafe.Pointer
    for k, v := range imgs {
        logo.Infoln("batch~~~~~~", k, " image: ", v)
        logo.Infoln("batch~~~~~~", k, " image: ", v, " size: ", wid, "x", hei, "x", chn)
        if v == nil {
            fake := C.create_fake_image(C.int(wid * hei * chn))
            freeFakeImage = append(freeFakeImage, fake)
            C.fill_images(cImgs, C.int(h.batchSize), C.int(k), fake, C.int(wid), C.int(hei), C.int(chn))
        } else {
            logo.Infoln("batch~~~~~~ real image len: ", len(v.Data), " size: ", v.Width, "x", v.Height, "x", v.Channel)
            C.fill_images(cImgs, C.int(h.batchSize), C.int(k), unsafe.Pointer(&v.Data[0]), C.int(v.Width), C.int(v.Height), C.int(v.Channel))
        }
    }