[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


延伸閱讀

1.
LEADERG APP 基礎人工智能軟體 (No Code ,不用寫程式, APP 操作介面, Deep Learning, 深度學習, Machine Learning, 機器學習, Big Data, 大數據)

2.
LEADERG AI ZOO 先進人工智能軟體 (Low Code ,提供豐富範例程式, Jupyter Lab 操作介面, AI Model Zoo, Deep Learning, 深度學習, Machine Learning, 機器學習, Big Data, 大數據)

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

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

購買方式

歡迎聯絡我們索取報價單,幫助您購買適合的產品。
客服信箱: leaderg@leaderg.com
客服電話: 02-2784-9788

感謝客戶們的支持

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