AI圖像修復

高清圖像修復 （python）

功能：超高分辨率圖像修復。
樣例輸入：待修復的jpg圖片以及對應的mask圖片。
樣例輸出：修復后的圖片。

前置條件

請檢查以下條件要求是否滿足，如不滿足請按照備注進行相應處理。如果CANN版本升級，請同步檢查第三方依賴是否需要重新安裝（5.0.4及以上版本第三方依賴和5.0.4以下版本有差異，需要重新安裝）。

樣例準備

1. 獲取源碼包。

   可以使用以下兩種方式下載，請選擇其中一種進行源碼準備。

   • 命令行方式下載（下載時間較長，但步驟簡單）。

     # 開發環境，非root用戶命令行中執行以下命令下載源碼倉。    
     cd ${HOME}     
     git clone https://gitee.com/ascend/samples.git

      
     注：如果需要切換到其它tag版本，以v0.5.0為例，可執行以下命令。

     git checkout v0.5.0

      
   • 壓縮包方式下載（下載時間較短，但步驟稍微復雜）。
     注：如果需要下載其它版本代碼，請先請根據前置條件說明進行samples倉分支切換。

      # 1. samples倉右上角選擇 【克隆/下載】 下拉框并選擇 【下載ZIP】。    
      # 2. 將ZIP包上傳到開發環境中的普通用戶家目錄中，【例如：${HOME}/ascend-samples-master.zip】。     
      # 3. 開發環境中，執行以下命令，解壓zip包。     
      cd ${HOME}    
      unzip ascend-samples-master.zip

      

2. 獲取此應用中所需要的模型并轉換。

   模型名稱	模型說明	模型下載路徑
   image_inpainting	使用ImageInpainting模型對超高分辨率圖像修復	image_inpainting：https://gitee.com/ascend/ModelZoo-TensorFlow/tree/master/TensorFlow/contrib/cv/imageinpainting_HiFill

   cd ${HOME}/samples/python/level2_simple_inference/6_other/imageinpainting_hifill/model
   wget https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/matmul_27648.json   
   wget https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/hifill.pb
   atc --output_type=FP32 --input_shape="img:1,512,512,3;mask:1,512,512,1" --input_format=NHWC --output="./hifill" --soc_version=Ascend310 --framework=3 --save_original_model=false --model="./hifill.pb"
   atc --singleop=./matmul_27648.json --output=./0_BatchMatMul_0_0_1_1_1024_1024_0_0_1_1_1024_27648_0_0_1_1_1024_27648 --soc_version=Ascend310

    

3. 獲取樣例需要的測試圖片。

   # 執行以下命令，下載對應的測試圖片。
   cd ${HOME}/samples/python/level2_simple_inference/6_other/imageinpainting_hifill/data
   wget https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/data/test.jpg
   cd ${HOME}/samples/python/level2_simple_inference/6_other/imageinpainting_hifill/mask
   wget https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/mask/test.jpg 
   cd ../src

    

樣例運行

注：開發環境與運行環境合一部署，請跳過步驟1，直接執行步驟2即可。

1. 執行以下命令,將開發環境的 imageinpainting_hifill 目錄上傳到運行環境中，例如 /home/HwHiAiUser，并以HwHiAiUser（運行用戶）登錄運行環境（Host）。

   # 【xxx.xxx.xxx.xxx】為運行環境ip，200DK在USB連接時一般為192.168.1.2，300（ai1s）為對應的公網ip。
   scp -r $HOME/samples/python/level2_simple_inference/6_other/imageinpainting_hifill HwHiAiUser@xxx.xxx.xxx.xxx:/home/HwHiAiUser
   ssh HwHiAiUser@xxx.xxx.xxx.xxx
   cd ${HOME}/imageinpainting_hifill/src    

    
2. 運行可執行文件。

    python3.6 main.py

    

查看結果

運行完成后，會在out目錄下生成帶推理結果的jpg圖片。

常見錯誤

請參考常見問題定位對遇到的錯誤進行排查。如果wiki中不包含，請在samples倉提issue反饋。