256 lines
12 KiB
Python
256 lines
12 KiB
Python
|
import configparser
|
|||
|
|
import math
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import numpy as np
|
|||
|
|
import cv2
|
|||
|
|
|
|||
|
|
from PyQt5 import QtWidgets
|
|||
|
|
from PyQt5.QtCore import QThread, pyqtSignal
|
|||
|
|
from PyQt5.QtWidgets import QApplication, QHeaderView, QTableWidgetItem, QFileDialog
|
|||
|
|
|
|||
|
|
from Creat_Window import Ui_MainWindow
|
|||
|
|
|
|||
|
|
class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
|
|||
|
|
def __init__(self):
|
|||
|
|
QtWidgets.QMainWindow.__init__(self) # 創建主界面對象
|
|||
|
|
self.setupUi(self)
|
|||
|
|
self.database_create = Database_Create()
|
|||
|
|
self.database_create.Created_Signal.connect(self.Created)
|
|||
|
|
self.messages_Finish_count = 0
|
|||
|
|
self.tableWidget.setColumnCount(3)
|
|||
|
|
self.tableWidget.setRowCount(16)
|
|||
|
|
self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
|
|||
|
|
self.tableWidget.horizontalHeader().setSectionResizeMode(0, QHeaderView.Interactive)
|
|||
|
|
self.comboBox_ClassNames.addItems(folder_name for folder_name in os.listdir(self.database_create.input_folder))
|
|||
|
|
self.comboBox_IncreaseMultiplier.addItems([str(i) for i in range(1, 80)])
|
|||
|
|
|
|||
|
|
config = configparser.ConfigParser()
|
|||
|
|
config.read('Creat_Config.ini')
|
|||
|
|
self.label_InputFilePath.setText(config['FilePathInformation']['inputfilepath'])
|
|||
|
|
self.label_OutputFilePath.setText(config['FilePathInformation']['outputfilepath'])
|
|||
|
|
|
|||
|
|
self.initTableWidget()
|
|||
|
|
|
|||
|
|
self.iniGuiEvent()
|
|||
|
|
def iniGuiEvent(self):
|
|||
|
|
self.comboBox_ClassNames.currentIndexChanged.connect(self.ValueChange_ClassNames)
|
|||
|
|
self.comboBox_IncreaseMultiplier.currentIndexChanged.connect(self.ValueChange_IncreaseMultiplier)
|
|||
|
|
self.pushButton.clicked.connect(self.Creating)
|
|||
|
|
self.pushButton_ChangeInputFilePath.clicked.connect(self.ChangeFilePath)
|
|||
|
|
self.pushButton_ChangeOutputFilePath.clicked.connect(self.ChangeFilePath)
|
|||
|
|
def Creating(self):
|
|||
|
|
self.database_create.start()
|
|||
|
|
def Created(self, classname):
|
|||
|
|
item = QTableWidgetItem(str(len(os.listdir(os.path.join(self.database_create.output_folder, classname)))))
|
|||
|
|
self.tableWidget.setItem([cn for cn in self.database_create.ClassNameAndQuantity.keys()].index(classname.lower()), 2, item)
|
|||
|
|
|
|||
|
|
def initTableWidget(self):
|
|||
|
|
self.tableWidget.clear()
|
|||
|
|
row = 0
|
|||
|
|
for key in self.database_create.Config_dic.keys():
|
|||
|
|
# Write the value to the current (row, col) position in the TableWidget
|
|||
|
|
item = QTableWidgetItem(key)
|
|||
|
|
self.tableWidget.setItem(row, 0, item)
|
|||
|
|
item = QTableWidgetItem(str(self.database_create.Config_dic[key]))
|
|||
|
|
self.tableWidget.setItem(row, 1, item)
|
|||
|
|
if self.database_create.OutputFileFolderAmountList == []:
|
|||
|
|
item = QTableWidgetItem('None')
|
|||
|
|
else:
|
|||
|
|
item = QTableWidgetItem(str(self.database_create.OutputFileFolderAmountList[row]))
|
|||
|
|
self.tableWidget.setItem(row, 2, item)
|
|||
|
|
row += 1
|
|||
|
|
|
|||
|
|
def ValueChange_ClassNames(self):
|
|||
|
|
self.comboBox_IncreaseMultiplier.setCurrentIndex(int(self.database_create.Config_dic[self.comboBox_ClassNames.currentText().lower()])-1)
|
|||
|
|
def ValueChange_IncreaseMultiplier(self):
|
|||
|
|
self.database_create.Config_dic[self.comboBox_ClassNames.currentText().lower()] = int(self.comboBox_IncreaseMultiplier.currentText())
|
|||
|
|
self.initTableWidget()
|
|||
|
|
def ChangeFilePath(self):
|
|||
|
|
sender = self.sender() # Get the button that was clicked
|
|||
|
|
name = sender.accessibleName() # Get the accessibleName property
|
|||
|
|
print(f"Clicked button name: {name}")
|
|||
|
|
|
|||
|
|
dir = QFileDialog.getExistingDirectory(self, "Open Directory",
|
|||
|
|
'./',
|
|||
|
|
QFileDialog.ShowDirsOnly
|
|||
|
|
| QFileDialog.DontResolveSymlinks)
|
|||
|
|
if dir is not '':
|
|||
|
|
config = configparser.ConfigParser()
|
|||
|
|
config.read('Creat_Config.ini')
|
|||
|
|
if name == 'In':
|
|||
|
|
self.label_InputFilePath.setText(dir)
|
|||
|
|
self.database_create.input_folder = dir
|
|||
|
|
config.set('FilePathInformation', "inputfilepath", dir)
|
|||
|
|
config.write(open('Creat_Config.ini', "r+")) # r+模式
|
|||
|
|
elif name == 'Out':
|
|||
|
|
self.label_OutputFilePath.setText(dir)
|
|||
|
|
self.database_create.output_folder = dir
|
|||
|
|
config.set('FilePathInformation', "outputfilepath", dir)
|
|||
|
|
config.write(open('Creat_Config.ini', "r+")) # r+模式
|
|||
|
|
else:
|
|||
|
|
print('dir is None')
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Database_Create(QThread):
|
|||
|
|
Created_Signal = pyqtSignal(str)
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__()
|
|||
|
|
self.input_folder = ''
|
|||
|
|
self.output_folder = ''
|
|||
|
|
|
|||
|
|
self.ScaleROI_Data = []
|
|||
|
|
# self.ReadScaleROI_Data()
|
|||
|
|
|
|||
|
|
self.Config_dic = {}
|
|||
|
|
self.ClassNameAndQuantity = {}
|
|||
|
|
self.OutputFileFolderAmountList = []
|
|||
|
|
self.ReadINI()
|
|||
|
|
self.GettingClassInformation()
|
|||
|
|
self.Calculating()
|
|||
|
|
self.readOutputFilefolderAmonut()
|
|||
|
|
|
|||
|
|
def Unsharpen(self, inputimage):
|
|||
|
|
# USM锐化增强方法(Unsharpen Mask)
|
|||
|
|
# 先对原图高斯模糊，用原图减去系数x高斯模糊的图像
|
|||
|
|
# 再把值Scale到0~255的RGB像素范围
|
|||
|
|
# 优点：可以去除一些细小细节的干扰和噪声，比卷积更真实
|
|||
|
|
# （原图像-w*高斯模糊）/（1-w）；w表示权重（0.1~0.9），默认0.6
|
|||
|
|
|
|||
|
|
# sigma = 5、15、25
|
|||
|
|
blur_img = cv2.GaussianBlur(inputimage, (0, 0), 5)
|
|||
|
|
usm = cv2.addWeighted(inputimage, 1.5, blur_img, -0.5, 0)
|
|||
|
|
# cv.addWeighted(图1,权重1, 图2, 权重2, gamma修正系数, dst可选参数, dtype可选参数)
|
|||
|
|
|
|||
|
|
h, w = inputimage.shape[:2]
|
|||
|
|
result = np.zeros([h, w * 2], dtype=inputimage.dtype)
|
|||
|
|
result[0:h, 0:w] = inputimage
|
|||
|
|
result[0:h, w:2 * w] = usm
|
|||
|
|
# cv.putText(图像名，标题，（x坐标，y坐标），字体，字的大小，颜色，字的粗细）
|
|||
|
|
return usm
|
|||
|
|
|
|||
|
|
def ReadScaleROI_Data(self):
|
|||
|
|
with open(r"C:\Users\user\Desktop\ROI\Config\0_rotate_and_mirror.txt", 'r') as file:
|
|||
|
|
lines = file.readlines()
|
|||
|
|
for line in lines:
|
|||
|
|
line_data = line.strip().split(' ')
|
|||
|
|
if len(line_data) == 5: # 确保每行都有5个数据项
|
|||
|
|
line_data[1] = int(float(line_data[1]) * 3400)
|
|||
|
|
line_data[2] = int(float(line_data[2]) * 1852)
|
|||
|
|
self.ScaleROI_Data.append(tuple(map(int, line_data[0:3])))
|
|||
|
|
def ReadINI(self):
|
|||
|
|
config = configparser.ConfigParser()
|
|||
|
|
config.read('Creat_Config.ini')
|
|||
|
|
self.input_folder = config['FilePathInformation']['inputfilepath']
|
|||
|
|
self.output_folder = config['FilePathInformation']['outputfilepath']
|
|||
|
|
|
|||
|
|
def GettingClassInformation(self):
|
|||
|
|
file_list = os.listdir(self.input_folder)
|
|||
|
|
for className in file_list:
|
|||
|
|
self.ClassNameAndQuantity[className.lower()] = len((os.listdir(os.path.join(self.input_folder, className))))
|
|||
|
|
|
|||
|
|
def Calculating(self):
|
|||
|
|
for className in self.ClassNameAndQuantity.keys():
|
|||
|
|
self.Config_dic[className] = 1 if round(1250/(self.ClassNameAndQuantity[className]*8))==0 else round(1250/(self.ClassNameAndQuantity[className]*8))
|
|||
|
|
def WriteINI(self):
|
|||
|
|
config = configparser.ConfigParser()
|
|||
|
|
config['Multiple'] = {}
|
|||
|
|
for key in self.Config_dic.keys():
|
|||
|
|
config['Multiple'][key] = str(self.Config_dic[str(key).lower()])
|
|||
|
|
with open('Creat_Config.ini', 'w') as f:
|
|||
|
|
config.write(f)
|
|||
|
|
def CaculateRoateAngle(self, forder_name):
|
|||
|
|
value = self.Config_dic[forder_name.lower()]
|
|||
|
|
print(f'int(value){int(value)}')
|
|||
|
|
angle = 360 / int(value)
|
|||
|
|
|
|||
|
|
return angle
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
self.new_creat()
|
|||
|
|
# self.creat()
|
|||
|
|
|
|||
|
|
def creat(self):
|
|||
|
|
folder_counter = 0
|
|||
|
|
for folder_name in os.listdir(self.output_folder):
|
|||
|
|
folder_path = os.path.join(self.output_folder, folder_name)
|
|||
|
|
Angle = self.CaculateRoateAngle(folder_name)
|
|||
|
|
if os.path.isdir(folder_path):
|
|||
|
|
output_subfolder = os.path.join(self.output_folder, folder_name)
|
|||
|
|
os.makedirs(output_subfolder, exist_ok=True)
|
|||
|
|
print(f"Reading images from folder: {folder_path}")
|
|||
|
|
for file_name in os.listdir(folder_path):
|
|||
|
|
file_path = os.path.join(folder_path, file_name)
|
|||
|
|
if os.path.isfile(file_path) and file_name.endswith(('.jpg', '.png', '.jpeg')):
|
|||
|
|
image = cv2.imread(file_path)
|
|||
|
|
center = (image.shape[1] // 2, image.shape[0] // 2)
|
|||
|
|
for angle in range(int(Angle), 360, int(Angle)):
|
|||
|
|
print(int(angle))
|
|||
|
|
rotation_matrix = cv2.getRotationMatrix2D(center, angle, 1.0)
|
|||
|
|
rotated_image = cv2.warpAffine(image, rotation_matrix, (image.shape[1], image.shape[0]))
|
|||
|
|
resized_image = cv2.resize(rotated_image, (224, 224))
|
|||
|
|
new_file_name = f"{folder_counter}_rotated_{angle}_{file_name}"
|
|||
|
|
new_file_path = os.path.join(output_subfolder, new_file_name)
|
|||
|
|
|
|||
|
|
# if file_name == '12_Phillips_1' or file_name == '13_Phillips_2':
|
|||
|
|
resized_image = self.Unsharpen(resized_image)
|
|||
|
|
|
|||
|
|
cv2.imwrite(new_file_path, resized_image)
|
|||
|
|
print(f"Saved rotated image: {new_file_path}")
|
|||
|
|
self.Created_Signal.emit(folder_name)
|
|||
|
|
folder_counter += 1
|
|||
|
|
# self.WriteINI()
|
|||
|
|
print('end')
|
|||
|
|
def new_creat(self):
|
|||
|
|
folder_counter = 0
|
|||
|
|
for folder_name in os.listdir(self.input_folder):
|
|||
|
|
folder_path = os.path.join(self.input_folder, folder_name)
|
|||
|
|
if os.path.isdir(folder_path):
|
|||
|
|
output_subfolder = os.path.join(self.output_folder, folder_name)
|
|||
|
|
os.makedirs(output_subfolder, exist_ok=True)
|
|||
|
|
print(f"Reading images from folder: {folder_path}")
|
|||
|
|
for file_name in os.listdir(folder_path):
|
|||
|
|
file_path = os.path.join(folder_path, file_name)
|
|||
|
|
if os.path.isfile(file_path) and file_name.endswith(('.jpg', '.png', '.jpeg')):
|
|||
|
|
image = cv2.imread(file_path)
|
|||
|
|
center = (image.shape[1] // 2, image.shape[0] // 2)
|
|||
|
|
for angle in range(0, 360, 90):
|
|||
|
|
rotation_matrix = cv2.getRotationMatrix2D(center, angle, 1.0)
|
|||
|
|
rotated_image = cv2.warpAffine(image, rotation_matrix, (image.shape[1], image.shape[0]))
|
|||
|
|
resized_image = cv2.resize(rotated_image, (224, 224))
|
|||
|
|
new_file_name = f"{folder_counter}_rotated_{angle}_{file_name}"
|
|||
|
|
new_file_path = os.path.join(output_subfolder, new_file_name)
|
|||
|
|
cv2.imwrite(new_file_path, resized_image)
|
|||
|
|
new_file_name = f"{folder_counter}_rotated_{angle}_flipx_{file_name}"
|
|||
|
|
new_file_path = os.path.join(output_subfolder, new_file_name)
|
|||
|
|
cv2.imwrite(new_file_path, cv2.flip(resized_image, 0))
|
|||
|
|
self.Created_Signal.emit(folder_name)
|
|||
|
|
folder_counter += 1
|
|||
|
|
# self.WriteINI()
|
|||
|
|
print('end')
|
|||
|
|
def readOutputFilefolderAmonut(self):
|
|||
|
|
parent_folder = self.output_folder
|
|||
|
|
# Initialize an empty list to store the folder names and file counts
|
|||
|
|
file_counts = []
|
|||
|
|
# Iterate through entries (files and folders) in the parent folder
|
|||
|
|
if os.path.exists(parent_folder):
|
|||
|
|
for entry in os.scandir(parent_folder):
|
|||
|
|
if entry.is_dir():
|
|||
|
|
folder_name = entry.name
|
|||
|
|
|
|||
|
|
# List all files in the child folder
|
|||
|
|
files_in_child_folder = [f.name for f in os.scandir(entry.path) if f.is_file()]
|
|||
|
|
|
|||
|
|
# Get the count of files in the child folder and append it to the list
|
|||
|
|
file_count = len(files_in_child_folder)
|
|||
|
|
file_counts.append(file_count)
|
|||
|
|
self.OutputFileFolderAmountList = file_counts
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
app = QApplication(sys.argv) # 固定的，PyQt5程式都需要QApplication對象。sys.argv是命令列參數清單，確保程式可以按兩下運行
|
|||
|
|
MyWindow = MainWindow() # 初始化
|
|||
|
|
MyWindow.show() # 將視窗控制項顯示在螢幕上
|
|||
|
|
sys.exit(app.exec_()) # 程式運行，sys.exit方法確保程式完整退出
|