一 、 torchvision
1. torchvision.transforms : 用于图像预处理
2. torchvision.models : 预训练好的模型供
3. torchvision.io :提供了视频、图片和文件的 IO 操作的功能
4. torchvision.ops : 提供了许多计算机视觉的特定操作
5. torchvision.utils : 提供了一些可视化的方法
二、PyTorchVideo
专注于视频理解工作的深度学习库。PytorchVideo 提供了加速视频理解研究所需的可重用、模块化和高效的组件
三、torchtext
可以方便的对文本进行预处理,例如截断补长、构建词表等
参考: