pytorch工具

作者: 张王李刘赵孙杨 | 来源:发表于2018-11-22 23:26 被阅读27次

文档

  • 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作为替代。
    该工具本身是为Keras设计的generator,我们使用pytorch的话只需要使用其中的部分函数就可以做到了。

    image-tools
  • imgaug: 又是一个数据扩增的工具箱,我对比了它和 image-tools中的3d elastic transform,觉得 image-tools更好用一些,这个写出来仅供参考。

    imgaug

相关文章

网友评论

    本文标题:pytorch工具

    本文链接:https://www.haomeiwen.com/subject/ksxdqqtx.html