Golang 无依赖提取图片主色调插件库 color-extractor 使用指南

color-extractor 是一个轻量级的 Go 库，用于从图像中提取主色调，无需任何外部依赖。下面我将详细介绍如何使用这个库。

安装

首先安装该库：

go get github.com/muesli/color-extractor

基本用法

从文件中提取主色调

package main

import (
	"fmt"
	"github.com/muesli/color-extractor"
	"image"
	_ "image/jpeg" // 导入以支持JPEG格式
	_ "image/png"  // 导入以支持PNG格式
	"os"
)

func main() {
	// 打开图像文件
	file, err := os.Open("example.jpg")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	// 解码图像
	img, _, err := image.Decode(file)
	if err != nil {
		panic(err)
	}

	// 提取主色调
	colors := colorextractor.Extract(img)
	
	// 打印提取到的颜色 (RGB格式)
	for _, color := range colors {
		r, g, b := color.RGB()
		fmt.Printf("主色调: R:%d G:%d B:%d 十六进制: #%02X%02X%02X\n", 
			r, g, b, r, g, b)
	}
}

从内存中的图像数据提取

package main

import (
	"bytes"
	"fmt"
	"github.com/muesli/color-extractor"
	"image"
	_ "image/jpeg"
	"io/ioutil"
)

func main() {
	// 读取图像数据到内存
	data, err := ioutil.ReadFile("example.jpg")
	if err != nil {
		panic(err)
	}

	// 解码内存中的图像数据
	img, _, err := image.Decode(bytes.NewReader(data))
	if err != nil {
		panic(err)
	}

	// 提取最多5种主色调
	colors := colorextractor.ExtractN(img, 5)
	
	for _, color := range colors {
		r, g, b := color.RGB()
		fmt.Printf("主色调: #%02X%02X%02X\n", r, g, b)
	}
}

高级用法

调整提取参数

package main

import (
	"fmt"
	"github.com/muesli/color-extractor"
	"image"
	_ "image/jpeg"
	"os"
)

func main() {
	file, err := os.Open("example.jpg")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	img, _, err := image.Decode(file)
	if err != nil {
		panic(err)
	}

	// 自定义提取参数
	options := colorextractor.Options{
		DistanceThreshold: 30, // 颜色距离阈值
		MinCount:          50, // 最小像素数
	}
	
	colors := colorextractor.ExtractWithOptions(img, options)
	
	for _, color := range colors {
		r, g, b := color.RGB()
		fmt.Printf("主色调: RGB(%d, %d, %d)\n", r, g, b)
	}
}

获取颜色占比信息

package main

import (
	"fmt"
	"github.com/muesli/color-extractor"
	"image"
	_ "image/jpeg"
	"os"
)

func main() {
	file, err := os.Open("example.jpg")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	img, _, err := image.Decode(file)
	if err != nil {
		panic(err)
	}

	// 提取颜色及其占比
	colorCounts := colorextractor.ExtractWithCount(img)
	totalPixels := img.Bounds().Dx() * img.Bounds().Dy()

	for color, count := range colorCounts {
		r, g, b := color.RGB()
		percentage := float64(count) / float64(totalPixels) * 100
		fmt.Printf("颜色: #%02X%02X%02X 占比: %.2f%%\n", 
			r, g, b, percentage)
	}
}

性能优化

对于大图像，可以先缩小尺寸以提高处理速度：

package main

import (
	"fmt"
	"github.com/muesli/color-extractor"
	"image"
	_ "image/jpeg"
	"os"
	"golang.org/x/image/draw"
)

func resizeImage(img image.Image, width int) image.Image {
	bounds := img.Bounds()
	height := (bounds.Dy() * width) / bounds.Dx()
	dst := image.NewRGBA(image.Rect(0, 0, width, height))
	draw.BiLinear.Scale(dst, dst.Bounds(), img, bounds, draw.Over, nil)
	return dst
}

func main() {
	file, err := os.Open("large_image.jpg")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	img, _, err := image.Decode(file)
	if err != nil {
		panic(err)
	}

	// 缩小图像到最大宽度300像素
	resized := resizeImage(img, 300)
	
	colors := colorextractor.Extract(resized)
	
	for _, color := range colors {
		r, g, b := color.RGB()
		fmt.Printf("主色调: #%02X%02X%02X\n", r, g, b)
	}
}

注意事项

1. 该库支持常见的图像格式，但需要导入相应的解码器（如 image/jpeg, image/png）
2. 对于大图像，建议先缩小尺寸以提高处理速度
3. 提取结果的质量取决于图像内容和参数设置

这个库简单易用，适合需要快速提取图像主色调的场景，如生成调色板、主题色分析等应用。