Go的PNG编码速度很慢

Question

我正在使用go-opencv从内置的网络摄像头获取帧。 从相机获取图像所需的时间约为50毫秒。 编码PNG的时间约为300毫秒。 编码JPEG的速度快了3倍，但仍然非常慢。

为什么运行这么慢？

注意 ：我已经针对NodeJS编写了类似的代码，在同一台机器上执行，并且在达到30fps且带有附加图像处理的情况下绝对没有问题。 对我来说，这消除了方程式中的硬件问题。

我的代码如下所示：

import (
    "fmt"
    "image/png"

    "github.com/lazywei/go-opencv/opencv"
)

camera := opencv.NewCameraCapture(0)
if camera == nil {
    panic("Unable to open camera.")
}
defer camera.Release()

for {
    if camera.GrabFrame() {
        img := camera.RetrieveFrame(1)
        if img != nil {
            frame := img.ToImage()
            buffer := new(bytes.Buffer)
            png.Encode(buffer, frame)
        } else {
            fmt.Println("Unable to capture frame")
        }
    }

}

Answer 1

禁用压缩可将计算机上的编码性能提高一个数量级。 如果您不想在标准库之外查找png包，那可能是一个开始。

我还尝试了BufferPool （Go 1.9中的新增功能），但是与使用零BufferPool的编码器相比，它没有什么区别。 也许我做错了。 不幸的是，文档非常简洁。

package main

import (
    "bytes"
    "image"
    "image/png"
    "os"
    "testing"
)

func BenchmarkPNG_Encode(b *testing.B) {
    img, buf := loadImage(b)

    b.ResetTimer()

    for i := 0; i < b.N; i++ {
            buf.Reset()
            png.Encode(buf, img)
    }
}

func BenchmarkPNG_Encoder(b *testing.B) {
    img, buf := loadImage(b)
    enc := &png.Encoder{}

    b.ResetTimer()

    for i := 0; i < b.N; i++ {
            buf.Reset()
            enc.Encode(buf, img)
    }
}

func BenchmarkPNG_Encoder_NoCompression(b *testing.B) {
    img, buf := loadImage(b)
    enc := &png.Encoder{
            CompressionLevel: png.NoCompression,
    }

    b.ResetTimer()

    for i := 0; i < b.N; i++ {
            buf.Reset()
            enc.Encode(buf, img)
    }
}

func loadImage(b *testing.B) (image.Image, *bytes.Buffer) {
    // foo.png PNG 1920x1053 1920x1053+0+0 8-bit sRGB 251KB 0.000u 0:00.000
    f, err := os.Open("foo.png")
    if err != nil {
            b.Fatal(err)
    }

    img, err := png.Decode(f)
    if err != nil {
            b.Fatal(err)
    }
    f.Close()

    buf := &bytes.Buffer{}
    // grow the buffer once
    (&png.Encoder{CompressionLevel: png.NoCompression}).Encode(buf, img)

    return img, buf
}

再次，这是在我的机器上，具有大约1920x1080像素的图像-随机屏幕截图； 不知道这与照片有什么不同。 因人而异。

$ go test -v -bench . -benchmem 
goos: linux
goarch: amd64
BenchmarkPNG_Encode-8                         10         119289121 ns/op          884964 B/op         38 allocs/op
BenchmarkPNG_Encoder-8                        10         118001658 ns/op          884932 B/op         37 allocs/op
BenchmarkPNG_Encoder_NoCompression-8         100          13050664 ns/op          807156 B/op        212 allocs/op

有趣的是，没有压缩比使用压缩导致更多的分配。

Answer 2

为什么运行这么慢？

测量！ Go有内置的好分析工具，例如go test 。

PNG编码器很可能是纯的，“未优化”的Go。 如果这还不够快，那么您可能要考虑提供更快的PNG编码器。