一 、 torchvision

1. torchvision.transforms : 用于图像预处理

2. torchvision.models  : 预训练好的模型供

3. torchvision.io :提供了视频、图片和文件的 IO 操作的功能

4. torchvision.ops :  提供了许多计算机视觉的特定操作

5. torchvision.utils : 提供了一些可视化的方法

二、PyTorchVideo

专注于视频理解工作的深度学习库。PytorchVideo 提供了加速视频理解研究所需的可重用、模块化和高效的组件

三、torchtext

可以方便的对文本进行预处理,例如截断补长、构建词表等

参考:

datawhale 深入浅出pytorch

更多推荐