【原创】批量区分横屏竖屏照片
程序功能:
- 扫描指定目录下的所有图片文件(支持jpg、jpeg、png、gif格式)
- 根据图片宽高比判断是横屏(宽>高)还是竖屏(高≥宽)
- 在源目录下创建两个子文件夹:
landscape
(横屏)和portrait
(竖屏) - 将图片复制到对应的文件夹中
- 显示处理统计信息
使用方法:
- 将代码保存为
main.go
文件 - 打开命令行终端
- 运行命令:
go run main.go <图片目录路径>
go run main.go /path/to/your/photos
程序会在指定的目录下创建两个子文件夹,并将图片分类复制到对应的文件夹中。原始图片保持不变,所以您不用担心图片丢失。
package mainimport ("fmt""image"_ "image/gif"_ "image/jpeg"_ "image/png""io/fs""os""path/filepath""strings"
)func main() {// 检查命令行参数if len(os.Args) < 2 {fmt.Println("用法: go run main.go <图片目录路径>")return}// 获取图片目录路径sourcePath := os.Args[1]// 创建横屏和竖屏目录landscapePath := filepath.Join(sourcePath, "landscape")portraitPath := filepath.Join(sourcePath, "portrait")// 确保目标目录存在os.MkdirAll(landscapePath, os.ModePerm)os.MkdirAll(portraitPath, os.ModePerm)// 统计var totalCount, landscapeCount, portraitCount, errorCount int// 遍历目录下的所有文件err := filepath.WalkDir(sourcePath, func(path string, d fs.DirEntry, err error) error {if err != nil {return err}// 跳过目录if d.IsDir() {return nil}// 检查文件是否为图片ext := strings.ToLower(filepath.Ext(path))if ext != ".jpg" && ext != ".jpeg" && ext != ".png" && ext != ".gif" {return nil}// 跳过已经分类的图片if strings.Contains(path, landscapePath) || strings.Contains(path, portraitPath) {return nil}totalCount++// 打开图片文件file, err := os.Open(path)if err != nil {fmt.Printf("打开文件 %s 失败: %v\n", path, err)errorCount++return nil}defer file.Close()// 解码图片以获取尺寸img, _, err := image.DecodeConfig(file)if err != nil {fmt.Printf("解码图片 %s 失败: %v\n", path, err)errorCount++return nil}// 判断图片方向var targetDir stringif img.Width > img.Height {targetDir = landscapePathlandscapeCount++} else {targetDir = portraitPathportraitCount++}// 构建目标路径fileName := filepath.Base(path)targetPath := filepath.Join(targetDir, fileName)// 复制文件到目标目录err = copyFile(path, targetPath)if err != nil {fmt.Printf("复制文件 %s 到 %s 失败: %v\n", path, targetPath, err)errorCount++return nil}fmt.Printf("已处理: %s -> %s\n", path, targetPath)return nil})if err != nil {fmt.Printf("遍历目录失败: %v\n", err)return}// 打印统计信息fmt.Println("\n--- 处理完成 ---")fmt.Printf("总共处理图片: %d\n", totalCount)fmt.Printf("横屏图片: %d\n", landscapeCount)fmt.Printf("竖屏图片: %d\n", portraitCount)fmt.Printf("处理失败: %d\n", errorCount)
}// 复制文件
func copyFile(src, dst string) error {// 读取源文件data, err := os.ReadFile(src)if err != nil {return err}// 写入目标文件return os.WriteFile(dst, data, 0644)
}