pytorch 中的转换和target_transform之间的区别?



transform &pytorch中的Target_transform ?
上下文:
每个TorchVision数据集都包含两个参数:transform和target_transform,分别用于修改样本和标签。

如果你查看源代码,特别是任何torchvision Dataset类的__getitem__方法,例如,torchvision. datassets。,您可以看到transform和target_transform分别用于修改/增强/转换图像和目标。

这可能有用的例子包括对象检测和语义分割,如果你在源图像上应用平移/旋转/剪切/缩放/裁剪,你也会希望在边界框/分割蒙版上进行相应的变换。

作为一个额外的例子,您可以看看这个官方教程,其中使用target_transform将整数类标签转换为用于图像分类的one-hot格式。

transform用于转换和增强输入数据和特征,而target_transform用于操作与这些标签相关的标签,如一个热编码等。

最新更新