mirror of
https://github.com/LdDl/go-darknet.git
synced 2025-09-26 19:51:27 +08:00
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
package darknet
|
|
|
|
// #include <stdlib.h>
|
|
// #include "image.h"
|
|
import "C"
|
|
import (
|
|
"image"
|
|
"unsafe"
|
|
|
|
"golang.org/x/image/draw"
|
|
)
|
|
|
|
// DarknetImage represents the image buffer.
|
|
type DarknetImage struct {
|
|
Width int
|
|
Height int
|
|
ans []float32
|
|
image C.image
|
|
}
|
|
|
|
// Close and release resources.
|
|
func (img *DarknetImage) Close() error {
|
|
C.free_image(img.image)
|
|
img.ans = nil
|
|
return nil
|
|
}
|
|
|
|
// https://stackoverflow.com/questions/33186783/get-a-pixel-array-from-from-golang-image-image/59747737#59747737
|
|
func imgTofloat32(src image.Image) []float32 {
|
|
bounds := src.Bounds()
|
|
width, height := bounds.Max.X, bounds.Max.Y
|
|
srcRGBA := image.NewRGBA(src.Bounds())
|
|
draw.Copy(srcRGBA, image.Point{}, src, src.Bounds(), draw.Src, nil)
|
|
|
|
red := make([]float32, 0, width*height)
|
|
green := make([]float32, 0, width*height)
|
|
blue := make([]float32, 0, width*height)
|
|
for y := 0; y < height; y++ {
|
|
for x := 0; x < width; x++ {
|
|
idxSource := (y*width + x) * 4
|
|
pix := srcRGBA.Pix[idxSource : idxSource+4]
|
|
rpix, gpix, bpix := float32(pix[0])/257.0, float32(pix[1])/257.0, float32(pix[2])/257.0
|
|
red = append(red, rpix)
|
|
green = append(green, gpix)
|
|
blue = append(blue, bpix)
|
|
}
|
|
}
|
|
srcRGBA = nil
|
|
|
|
ans := make([]float32, len(red)+len(green)+len(blue))
|
|
copy(ans[:len(red)], red)
|
|
copy(ans[len(red):len(red)+len(green)], green)
|
|
copy(ans[len(red)+len(green):], blue)
|
|
red = nil
|
|
green = nil
|
|
blue = nil
|
|
return ans
|
|
}
|
|
|
|
// Image2Float32 Returns []float32 representation of image.Image
|
|
func Image2Float32(img image.Image) (*DarknetImage, error) {
|
|
// ans := imgTofloat32(img)
|
|
width := img.Bounds().Dx()
|
|
height := img.Bounds().Dy()
|
|
imgDarknet := &DarknetImage{
|
|
Width: width,
|
|
Height: height,
|
|
ans: imgTofloat32(img),
|
|
image: C.make_image(C.int(width), C.int(height), 3),
|
|
}
|
|
C.fill_image_f32(&imgDarknet.image, C.int(width), C.int(height), 3, (*C.float)(unsafe.Pointer(&imgDarknet.ans[0])))
|
|
return imgDarknet, nil
|
|
}
|
|
|
|
// Float32ToDarknetImage Converts []float32 to darknet image
|
|
func Float32ToDarknetImage(flatten []float32, width, height int) (*DarknetImage, error) {
|
|
imgDarknet := &DarknetImage{
|
|
Width: width,
|
|
Height: height,
|
|
image: C.make_image(C.int(width), C.int(height), 3),
|
|
}
|
|
C.fill_image_f32(&imgDarknet.image, C.int(width), C.int(height), 3, (*C.float)(unsafe.Pointer(&flatten[0])))
|
|
return imgDarknet, nil
|
|
}
|