122 lines
4.6 KiB
Python
122 lines
4.6 KiB
Python
'''
|
||
1、此版本所需文件：dn影像和asd测定的积分球辐亮度曲线
|
||
2、处理过程：不需要扣除暗电流，因为dn影像在采集时已经扣除了暗电流；
|
||
1）通过辐亮度曲线生成定标文件
|
||
'''
|
||
import numpy as np
|
||
import pandas as pd
|
||
import sys, os, traceback
|
||
from library.image_reader_writer import ImageReaderWriter
|
||
from radiance_calibration_ui import Ui_MainWindow
|
||
from library.multithread import Worker
|
||
|
||
from PyQt5.QtWidgets import QMainWindow, QFileDialog, QApplication
|
||
from PyQt5.QtCore import Qt
|
||
|
||
|
||
class EnterWindow(QMainWindow, Ui_MainWindow):
|
||
def __init__(self, parent=None):
|
||
super(EnterWindow, self).__init__(parent)
|
||
self.setupUi(self)
|
||
|
||
self.radiance_calibration_object = RadianceCalibration()
|
||
|
||
# self.setWindowState(Qt.WindowMaximized) # 初始化时就最大化窗口
|
||
|
||
self.rad_bt.clicked.connect(self.select_rad)
|
||
self.dn_bt.clicked.connect(self.select_dn)
|
||
self.out_file_bt.clicked.connect(self.select_out_file)
|
||
|
||
self.operate_bt.clicked.connect(self.operate)
|
||
|
||
self.tmp_rad_file_path = None # 用于保存当前路径，下次打开就默认在此路径下
|
||
self.tmp_dn_file_path = None # 用于保存当前路径，下次打开就默认在此路径下
|
||
self.tmp_out_file_path = None # 用于保存当前路径，下次打开就默认在此路径下
|
||
|
||
def select_rad(self):
|
||
if self.tmp_rad_file_path == None:
|
||
rad_file_path = QFileDialog.getOpenFileName(self, '选择asd辐亮度文件', os.path.dirname(__file__))[0]
|
||
|
||
self.tmp_rad_file_path = os.path.dirname(rad_file_path)
|
||
elif self.tmp_rad_file_path is not None:
|
||
rad_file_path = QFileDialog.getOpenFileName(self, '选择asd辐亮度文件', self.tmp_rad_file_path)[0]
|
||
|
||
self.tmp_rad_file_path = os.path.dirname(rad_file_path)
|
||
|
||
self.radiance_calibration_object.asd_radiance_file_path = rad_file_path
|
||
self.rad_le.setText(rad_file_path)
|
||
|
||
def select_dn(self):
|
||
if self.tmp_dn_file_path == None:
|
||
dn_file_path = QFileDialog.getOpenFileName(self, '选择dn影像', os.path.dirname(__file__))[0]
|
||
|
||
self.tmp_dn_file_path = os.path.dirname(dn_file_path)
|
||
elif self.tmp_dn_file_path is not None:
|
||
dn_file_path = QFileDialog.getOpenFileName(self, '选择dn影像', self.tmp_dn_file_path)[0]
|
||
|
||
self.tmp_dn_file_path = os.path.dirname(dn_file_path)
|
||
|
||
self.radiance_calibration_object.dn_file_path = dn_file_path
|
||
self.dn_le.setText(dn_file_path)
|
||
|
||
def select_out_file(self):
|
||
if self.tmp_out_file_path == None:
|
||
out_file_path = QFileDialog.getSaveFileName(self, '选择输出文件路径和文件名', os.path.dirname(__file__))[0]
|
||
|
||
self.tmp_out_file_path = os.path.dirname(out_file_path)
|
||
elif self.tmp_out_file_path is not None:
|
||
out_file_path = QFileDialog.getSaveFileName(self, '选择输出文件路径和文件名', self.tmp_out_file_path)[0]
|
||
|
||
self.tmp_out_file_path = os.path.dirname(out_file_path)
|
||
|
||
self.radiance_calibration_object.out_file_path = out_file_path
|
||
self.out_file_le.setText(out_file_path)
|
||
|
||
def operate(self):
|
||
|
||
self.radiance_calibration_object.operate()
|
||
|
||
|
||
class RadianceCalibration():
|
||
def __init__(self):
|
||
# asd_radiance_file = os.path.dirname(__file__) + '\\corning_radance.txt'
|
||
# img_file_path = os.path.dirname(__file__) + '\\jfq_dn'
|
||
# out_file = os.path.dirname(__file__) + '\\jfq_dn_gain'
|
||
|
||
self.asd_radiance_file_path = None
|
||
self.dn_file_path = None
|
||
|
||
self.img_data = None
|
||
|
||
self.out_file_path = None
|
||
|
||
def operate(self):
|
||
try:
|
||
# 读取asd辐亮度数据
|
||
data = pd.read_csv(self.asd_radiance_file_path, sep='\t', dtype=np.float64, header=None)
|
||
self.asd_radiance = np.array(data)
|
||
|
||
# 读取影像
|
||
img_proj, img_geotrans, self.img_data = ImageReaderWriter.read_img(self.dn_file_path)
|
||
|
||
# 将影像所有行平均，得到一行（帧）影像
|
||
img_data_ave = np.mean(self.img_data, axis=1)
|
||
|
||
gain = np.empty((img_data_ave.shape[0], 1, img_data_ave.shape[1]))
|
||
for i in range(gain.shape[2]):
|
||
gain[:, 0, i] = self.asd_radiance[:, 1] / img_data_ave[:, i]
|
||
|
||
ImageReaderWriter.write_img(self.out_file_path, gain)
|
||
except:
|
||
traceback.print_exc()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
app = QApplication(sys.argv)
|
||
|
||
# 实例化主窗口
|
||
enter_window_instance = EnterWindow()
|
||
enter_window_instance.show()
|
||
|
||
sys.exit(app.exec_())
|