Question

我是golang的新手，我正在尝试获取目录中列出的所有图像的图像大小。这就是我所做的

package main

import (
    "fmt"
    "image"
    _ "image/jpeg"
    "io/ioutil"
    "os"
)

const dir_to_scan string = "/home/da/to_merge"

func main() {
    files, _ := ioutil.ReadDir(dir_to_scan)
    for _, filepath := range files {

        if reader, err := os.Open(filepath.Name()); err != nil {
            defer reader.Close()
            im, _, err := image.DecodeConfig(reader)
            if err != nil {
                fmt.Fprintf(os.Stderr, "%s: %v\n", filepath.Name(), err)
                continue
            }
            fmt.Printf("%s %d %d\n", filepath.Name(), im.Width, im.Height)
        } else {
            fmt.Println("Impossible to open the file")
        }
    }
}

我在image.DecodeConfig时遇到错误，即image: unknown format 有人知道正确的方法吗？在这里的文档中http://golang.org/src/pkg/image/format.go?s=2676:2730#L82说我应该传递io.Reader作为参数，这就是我正在做的事情。

Answer 1

您的代码存在两个问题。

第一个是您已将测试err != nil反转，因此在出现错误的情况下，您尝试解码图像。它应该是err == nil。

第二个，正如jimt所说的那样，你使用的filepath.Name()只包含os.Open()中的文件名，这就是err始终被设置的原因，因此始终输入if，并解码不存在的文件。

以下是更正后的代码：

package main

import (
    "fmt"
    "image"
    _ "image/jpeg"
    "io/ioutil"
    "os"
    "path/filepath"
)

const dir_to_scan string = "/home/da/to_merge"

func main() {
    files, _ := ioutil.ReadDir(dir_to_scan)
    for _, imgFile := range files {

        if reader, err := os.Open(filepath.Join(dir_to_scan, imgFile.Name())); err == nil {
            defer reader.Close()
            im, _, err := image.DecodeConfig(reader)
            if err != nil {
                fmt.Fprintf(os.Stderr, "%s: %v\n", imgFile.Name(), err)
                continue
            }
            fmt.Printf("%s %d %d\n", imgFile.Name(), im.Width, im.Height)
        } else {
            fmt.Println("Impossible to open the file:", err)
        }
    }
}

另外，如果您的目录中有其他图片格式，请不要忘记添加除image/jpeg之外的其他导入。

使用golang获取图像大小

1 个答案: