pytorch中transform和target_transform的区别？

pkmbmrz7 于 2023-10-20 发布在其他

关注(0)|答案(2)|浏览(194)

pytorch中transform和target_transform的区别？
内容：
每个TorchVision数据集都包含两个参数：transform和target_transform来分别修改样本和标签。

pytorch

来源：https://stackoverflow.com/questions/72852126/difference-between-transform-target-transform-in-pytorch

2条答案

按热度按时间

uhry853o1#

如果查看源代码，特别是torchvision Dataset classes的__getitem__方法，例如torchvision.datasets.DatasetFolder，您可以看到transform和target_transform分别用于修改/增强/转换图像和目标。
这可能有用的示例包括对象检测和语义分割，如果您在源图像上应用平移/旋转/剪切/缩放/裁剪，则还需要在边界框/分割蒙版上进行相应的变换。
作为一个额外的例子，你可以看看这个官方教程，其中target_transform用于将整数类标签转换为图像分类的独热格式。