LAPTOP-S9HJSOEB\昊天
10 months ago
parent
05ecb74903
commit
dbc376afd9
@ -0,0 +1,115 @@
|
||||
package com.zhehekeji.web.controller;
|
||||
|
||||
import java.awt.*;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.io.*;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.nio.file.StandardCopyOption;
|
||||
import java.util.*;
|
||||
import java.util.List;
|
||||
import javax.imageio.ImageIO;
|
||||
|
||||
public class BatchImageCropper {
|
||||
public static void moveRandomFiles(String sourceDir, String targetDir, int numFiles) throws Exception {
|
||||
File sourceFolder = new File(sourceDir);
|
||||
File[] files = sourceFolder.listFiles((dir, name) -> new File(dir, name).isFile());
|
||||
|
||||
if (files == null || files.length == 0) {
|
||||
System.out.println("源文件夹中没有文件可供移动。");
|
||||
return;
|
||||
}
|
||||
|
||||
List<File> fileList = Arrays.asList(files);
|
||||
Collections.shuffle(fileList); // 随机打乱文件列表
|
||||
|
||||
int moveCount = Math.min(numFiles, fileList.size());
|
||||
for (int i = 0; i < moveCount; i++) {
|
||||
File fileToMove = fileList.get(i);
|
||||
Path targetPath = Paths.get(targetDir, fileToMove.getName());
|
||||
Files.move(fileToMove.toPath(), targetPath, StandardCopyOption.REPLACE_EXISTING);
|
||||
// txt 文件名
|
||||
String txtFileName = fileToMove.getPath().replace("images","labels").replace(".BMP", ".txt");
|
||||
String txtPath = fileToMove.getPath().replace("train","val").replace("images","labels").replace(".BMP", ".txt");
|
||||
Files.move(Paths.get(txtFileName), Paths.get(txtPath), StandardCopyOption.REPLACE_EXISTING);
|
||||
System.out.println("Moved: " + fileToMove.getName());
|
||||
}
|
||||
}
|
||||
|
||||
public static void main2(String[] args) {
|
||||
try {
|
||||
String sourceDir = "D:\\PycharmProjects\\yolo\\gjg\\train\\images";
|
||||
String targetDir = "D:\\PycharmProjects\\yolo\\gjg\\val\\images";
|
||||
int numFiles = 30;
|
||||
|
||||
moveRandomFiles(sourceDir, targetDir, numFiles);
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
public static void main(String[] args) throws IOException {
|
||||
// 图片文件夹路径
|
||||
String imagesFolderPath = "D:\\WeChet\\WeChat Files\\wxid_ttkf0xgmyihv22\\FileStorage\\File\\2025-05\\新建文件夹\\献礼";
|
||||
// 标注文件夹路径
|
||||
String labelsFolderPath = "D:\\WeChet\\WeChat Files\\wxid_ttkf0xgmyihv22\\FileStorage\\File\\2025-05\\新建文件夹\\献礼\\labels";
|
||||
// 裁剪区域的坐标和尺寸
|
||||
int cropX = 950, cropY = 356, cropWidth = 1680, cropHeight = 1680;
|
||||
|
||||
// 获取所有图片文件
|
||||
File imagesFolder = new File(imagesFolderPath);
|
||||
File[] imageFiles = imagesFolder.listFiles((dir, name) -> name.endsWith(".BMP"));
|
||||
|
||||
// 对每张图片进行处理
|
||||
for (File imageFile : imageFiles) {
|
||||
// 读取图片
|
||||
BufferedImage image = ImageIO.read(imageFile);
|
||||
|
||||
// 裁剪图片
|
||||
// BufferedImage croppedImage = image.getSubimage(cropX, cropY, cropWidth, cropHeight);
|
||||
// // 直接覆盖原图片
|
||||
// ImageIO.write(croppedImage, "BMP", imageFile);
|
||||
|
||||
// 处理标注文件
|
||||
String labelFilePath = labelsFolderPath + "/" + imageFile.getName().replaceAll("\\.BMP$", ".txt");
|
||||
File labelFile = new File(labelFilePath);
|
||||
if (labelFile.exists()) {
|
||||
List<String> newLabels = adjustYoloLabels(labelFilePath, cropX, cropY, cropWidth, cropHeight,6);
|
||||
// 直接覆盖原标注文件
|
||||
BufferedWriter writer = new BufferedWriter(new FileWriter(labelFilePath));
|
||||
for (String line : newLabels) {
|
||||
writer.write(line);
|
||||
writer.newLine();
|
||||
}
|
||||
writer.close();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 修改YOLO标注坐标
|
||||
public static List<String> adjustYoloLabels(String labelFilePath, int cropX, int cropY, int cropWidth, int cropHeight,int classId) throws IOException {
|
||||
List<String> newLabels = new ArrayList<>();
|
||||
BufferedReader reader = new BufferedReader(new FileReader(labelFilePath));
|
||||
String line;
|
||||
|
||||
while ((line = reader.readLine()) != null) {
|
||||
String[] parts = line.split(" ");
|
||||
double centerX = Double.parseDouble(parts[1]);
|
||||
double centerY = Double.parseDouble(parts[2]);
|
||||
double width = Double.parseDouble(parts[3]);
|
||||
double height = Double.parseDouble(parts[4]);
|
||||
|
||||
// 计算裁剪后的相对坐标
|
||||
centerX = (centerX * 3072 - cropX) / cropWidth;
|
||||
centerY = (centerY * 2048 - cropY) / cropHeight;
|
||||
width = width * cropWidth / cropWidth;
|
||||
height = height * cropHeight / cropHeight;
|
||||
|
||||
// 更新YOLO标注格式
|
||||
newLabels.add(classId + " " + centerX + " " + centerY + " " + width + " " + height);
|
||||
}
|
||||
|
||||
reader.close();
|
||||
return newLabels;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue