學習pytorch: 語義分割工具

文檔

• Awesome-Pytorch-list: 提供了pytorch的各種工具，代碼，文檔等，下面的2,3是在裏面挑出來我使用過的工具。

可視化

• tensorboardX: tensorboard的擴展，可以結合pytorch使用，很簡單。
  

數據擴增

• Augmentor: 圖像增強工具箱，很強大。
  

• image-tools: 論文u-net和3d-unet中都提到了elastic deformation的數據擴增方法，在二維圖像中建議使用Augmentor，但是在三維圖像的augmentaiton上，Augmentor就不支持了，因此找到了這個image-tools作爲替代。
  該工具本身是爲Keras設計的generator，我們使用pytorch的話只需要使用其中的部分函數就可以做到了。
  

• imgaug: 又是一個數據擴增的工具箱，我對比了它和 image-tools中的3d elastic transform，覺得 image-tools更好用一些，這個寫出來僅供參考。