文档
- Awesome-Pytorch-list: 提供了pytorch的各种工具,代码,文档等,下面的2,3是在里面挑出来我使用过的工具。
可视化
-
tensorboardX: tensorboard的扩展,可以结合pytorch使用,很简单。
tensorboardX
数据扩增
-
Augmentor: 图像增强工具箱,很强大。
Augmentor
-
image-tools: 论文u-net和3d-unet中都提到了elastic deformation的数据扩增方法,在二维图像中建议使用Augmentor,但是在三维图像的augmentaiton上,Augmentor就不支持了,因此找到了这个image-tools作为替代。
image-tools
该工具本身是为Keras设计的generator,我们使用pytorch的话只需要使用其中的部分函数就可以做到了。
-
imgaug: 又是一个数据扩增的工具箱,我对比了它和 image-tools中的3d elastic transform,觉得 image-tools更好用一些,这个写出来仅供参考。
imgaug
网友评论