Pytorch数据处理工具箱

Pytorch数据处理工具箱

 torchvision

 transforms提供了对PIL Image对象和Tensor对象的常用操作。

1）对PIL Image的常见操作如下。

 

 transforms提供了对PIL Image对象和Tensor对象的常用操作。

2）对Tensor的常见操作如下。

 如果要对数据集进行多个操作，可通过Compose将这些操作像管道一样拼接起来，类似于nn.Sequential。

 

ImageFolder可以读取不同目录下的图像数据。

 

可视化工具

 TensorBoard简介

使用TensorBoard的一般步骤如下。
1）导入tensorboard，实例化SummaryWriter类，指明记录日志路径等信息。

 使用TensorBoard的一般步骤如下。
2）调用相应的API接口，接口一般格式为：
add_xxx(tag_name, object, iteration)
其中，xxx指的是各种可视化方法。