| | |
| | | |
| | | 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; |
| | | } |
| | |
| | | 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)) |
| | | } |
| | | } |