[LEADERG AI ZOO] Jupyter-Image-Pix2Pix-PyTorch

[簡介]

 

 

Pix2Pix 風格轉換。將原影像輸出訓練學習的影像風格。例如 : 黑白轉換成彩色、邊緣圖轉換相片、地圖轉換成衛星地圖等各種應用。

 

 

 

[操作步驟及說明]

 

 

1_combine_A_and_B.ipynb

 

data/A 與 data/B 資料夾內有 train、val、test ,A 是放置轉換後的影像、B 是放置轉換前的影像。

 

請將影像依照轉換前後放入 A、B 資料夾中的 train、val、test,需注意一點的是 兩資料夾中的 train、val、test 皆需要能夠找到對應的影像檔名。

 

接著按下 1. combine A and B 將對應的影像合併為一張,且輸出至 data/train, data/val, data/test。

 

注意 : 

1). 影像長寬建議相同,正方形,如果不相同可以縮放或裁切等方式。

2). 圖檔與其附檔名需為 .jpg。

3). 影像需兩兩相對應。    

 

 

2_visdom_server.ipynb

 

開啟 port 為 8801 的 visdom_server。

 

 

3_visdom_server_browser.ipynb

 

開啟瀏覽器,可看到訓練過程的 loss 曲線圖等資訊。

 

 

4_train.ipynb

 

訓練模型。

 

•    --continue_train : 載入 model/latest 模型接續訓練。

 

•   --epoch_count 2300 : 輸出模型檔名從 2300 次開始,表示前面已經有訓練 2300 期。與 continue_train 搭配使用。

 

•    --n_epochs 1000 : 訓練期數 1000。

 

•    --gpu_ids 0 : 使用哪顆 gpu 訓練,也可設定多顆,例如  --gpu_ids 0,1,2。

 

•    --save_epoch_freq 5 : 儲存模型頻率。

 

 

5_inference_folder.ipynb

 

推論資料夾。

 

•    checkpoints_dir = 'model' : 模型資料夾。

 

•    epoch = 'latest' : 載入 latest_net_G_A 的模型。

 

•    dataroot = 'data/B'、phase = 'test' : 測試資料夾為 data/B/test。

 

•    num_test = 27 : 資料夾中有多少張測試影像。

 

 

6_inference_folder_image_pairs.ipynb

 

推論資料夾,資料夾內的影像為 1_combine_A_and_B.ipynb 合併 A、B 的影像。

可以用來比照目標與模型生成的影像。

 

•    checkpoints_dir = 'model'、epoch = 'latest' : 模型為 model/latest_net_G_A.pth。

 

•    dataroot = 'data/'、phase = 'test' : 測試資料夾為 data/test。

 

•    num_test = 20 : 資料夾中有多少張測試影像。

 

 

 

推論結果.png

聯絡我們及購買方式

請參閱以下網址:
https://tw.leaderg.com/article/index?sn=11056

延伸閱讀

1.
LEADERG App4AI 人工智能軟體 (App for AI, 不用寫程式, 視窗操作介面, 易學易用, 無經驗可, AI Model Zoo, Deep Learning, 深度學習, Machine Learning, 機器學習, 人工智慧)

2.
LEADERG AI Zoo 人工智能演算法 (整理及優化190種範例程式碼, 節省90%的開發時間, 10倍速快速開發 AI 程式, Jupyter Lab 操作介面, Code for AI, Code4AI, Less Coding, AI Model Zoo, 深度學習, 機器學習, Big Data, 大數據)

3.
LEADERG ACE 人工智能電腦 (工作站、伺服器、筆電,GPU, computer, workstation, notebook, server, NVIDIA, RTX-3090-24G, RTX-A6000-48G, A100-40G, deep learning, 深度學習, machine learning, 機器學習)

4.
LEADERG AI Education 人工智能教育 (AI 教育, 影像分析, 影像分類, 物件偵測, 影像切割, 資料分析, deep learning, 深度學習, machine learning, 機器學習, big data, 大數據)

感謝客戶們的支持

臺灣大學、清華大學、陽明交通大學、成功大學、臺北醫學大學、臺北護理健康大學、中興大學、暨南大學、宜蘭大學、聯合大學、淡江大學、逢甲大學、長庚大學、義守大學、實踐大學、大同大學、臺科大、北科大、臺中科大、雲林科大、勤益科大、虎尾科大、屏東科大、高雄科大、朝陽科大、明志科大、南臺科大、亞東科大、中國科大、國防大學、海軍軍官學校、陸軍專科學校、鼓山高中、臺大醫院、榮民總醫院、長庚醫院、慈濟醫院、義大醫院、中山科學研究院、國防部軍備局、法務部調查局、工研院、資策會、原子能委員會核能研究所、農業委員會特有生物研究保育中心、勞動部勞動研究所、紡織研究所、金屬工業研究發展中心、台灣儀器科技研究中心、車輛測試中心、台灣自來水公司、台積電、聯電、南亞科、華邦電子、精材科技、華信光電、友達光電、群創光電、瀚宇彩晶、台塑、台塑網、南亞塑膠、台灣化學纖維、中國石油、羅技電子、義隆電子、隆達電子、達方電子、鴻海、大聯大 - 世平興業、盟立自動化、迅得機械、南茂科技、順達科技、致伸科技、英特盛、豐興鋼鐵、中宇環保工程、竹陞科技、研揚科技、敦陽科技、竑騰科技、先豐通訊、神達電腦、神通電腦、百佳泰、科音國際、峰安車業、鴻鵠國際、藍海智能等。