Firsh Push at 20241207
This commit is contained in:
62
Image_Process/load_and_ImageGenerator.py
Normal file
62
Image_Process/load_and_ImageGenerator.py
Normal file
@@ -0,0 +1,62 @@
|
||||
from Load_process.LoadData import Loding_Data_Root
|
||||
from Image_Process.Image_Generator import Image_generator
|
||||
from Load_process.file_processing import Process_File
|
||||
from model_data_processing.processing_for_cut_image import Cut_Indepentend_Data
|
||||
from Load_process.Loading_Tools import Load_Data_Prepare, Load_Data_Tools
|
||||
|
||||
class Load_ImageGenerator():
|
||||
'''
|
||||
這是一個拿來進行資料強化的物件,最主要結合了學姊給的資料強化與我自行設定的資料強化。
|
||||
藉由此物件先將資料讀取出來,並將資料分別進行資料強化,利用資料強化來迷部資料的不平衡
|
||||
這只是其中一個實驗
|
||||
|
||||
Parmeter
|
||||
standard_root: 做跟學姊給的資料強化同一種的資料強化
|
||||
myself_root: 資料強化的內容參數是我自己設定的
|
||||
IndependentDataRoot: 要存回去的資料夾路徑
|
||||
Herpeslabels: 皰疹的類別
|
||||
MonKeyPoxlabels: 猴痘的類別(猴痘、水痘、正常)
|
||||
herpes_data: 合併herpes Dataset的資料成一個List
|
||||
MonkeyPox_data: 合併MonkeyPox DataSet 的資料成一個List
|
||||
'''
|
||||
def __init__(self, Training_Root,Test_Root, Validation_Root, Generator_Root, Labels) -> None:
|
||||
self.Training_Root = Training_Root
|
||||
self.TestRoot = Test_Root
|
||||
self.ValidationRoot = Validation_Root
|
||||
self.GeneratoRoot = Generator_Root
|
||||
self.Labels = Labels
|
||||
pass
|
||||
|
||||
def process_main(self, Data_Length : int):
|
||||
File = Process_File()
|
||||
Prepare = Load_Data_Prepare()
|
||||
load = Loding_Data_Root(self.Labels, self.Training_Root, self.GeneratoRoot)
|
||||
Indepentend = Cut_Indepentend_Data(self.Training_Root, self.Labels)
|
||||
Load_Tool = Load_Data_Tools()
|
||||
Generator = Image_generator(self.GeneratoRoot, self.Labels)
|
||||
|
||||
# 將測試資料獨立出來
|
||||
test_size = 0.1
|
||||
Indepentend.IndependentData_main(self.TestRoot, test_size)
|
||||
|
||||
# 將驗證資料獨立出來
|
||||
test_size = 0.1
|
||||
Indepentend.IndependentData_main(self.ValidationRoot, test_size)
|
||||
|
||||
if not File.Judge_File_Exist(self.GeneratoRoot): # 檔案若不存在
|
||||
# 確定我要多少個List
|
||||
Prepare.Set_Data_Content([], Data_Length)
|
||||
|
||||
# 製作讀檔字典並回傳檔案路徑
|
||||
Prepare.Set_Label_List(self.Labels)
|
||||
Prepare.Set_Data_Dictionary(Prepare.Get_Label_List(), Prepare.Get_Data_Content(), Data_Length)
|
||||
Original_Dict_Data_Root = Prepare.Get_Data_Dict()
|
||||
get_all_original_image_data = Load_Tool.get_data_root(self.Training_Root, Original_Dict_Data_Root, Prepare.Get_Label_List())
|
||||
|
||||
# 儲存資料強化後資料
|
||||
Generator.Processing_Main(get_all_original_image_data) # 執行資料強化
|
||||
else: # 若檔案存在
|
||||
print("standard data and myself data are exist\n")
|
||||
|
||||
# 執行讀檔
|
||||
return load.process_main()
|
||||
Reference in New Issue
Block a user