mirror of
https://github.com/LdDl/go-darknet.git
synced 2025-09-26 19:51:27 +08:00
7e34b459e6
imgTofloat32 + fill_image_f32 took more than double the time since tofloat created needless allocations and iterate over each pixel, and fill_image in c then iterate over each pixel*3 again just to move it into the c image.
57 lines
1.4 KiB
Go
57 lines
1.4 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
|
|
image C.image
|
|
}
|
|
|
|
// Close and release resources.
|
|
func (img *DarknetImage) Close() error {
|
|
C.free_image(img.image)
|
|
return nil
|
|
}
|
|
|
|
func Image2Float32(src image.Image) (*DarknetImage, error) {
|
|
bounds := src.Bounds()
|
|
srcRGBA := image.NewRGBA(bounds)
|
|
draw.Copy(srcRGBA, image.Point{}, src, bounds, draw.Src, nil)
|
|
|
|
return ImageRGBA2Float32(srcRGBA)
|
|
}
|
|
|
|
// Image2Float32 Returns []float32 representation of image.Image
|
|
func ImageRGBA2Float32(img *image.RGBA) (*DarknetImage, error) {
|
|
width := img.Bounds().Dx()
|
|
height := img.Bounds().Dy()
|
|
imgDarknet := &DarknetImage{
|
|
Width: width,
|
|
Height: height,
|
|
image: C.make_image(C.int(width), C.int(height), 3),
|
|
}
|
|
C.to_float_and_fill_image(&imgDarknet.image, C.int(width), C.int(height), (*C.uint8_t)(unsafe.Pointer(&img.Pix[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
|
|
}
|