关于totensor的问题

DLLXW / data-science-competition

该仓库用于记录作者本人参加的各大数据科学竞赛的获奖方案源码以及一些新比赛的原创baseline. 主要涵盖：kaggle, 阿里天池，华为云大赛校园赛，百度aistudio，和鲸社区，datafountain等

1.29k stars 476 forks source link

Open ghost opened 3 years ago

ghost commented 3 years ago

您好，谢谢您的分享！请问在data-science-competition/kaggle/Cassava Leaf Disease Classification/里，数据增强用的是ToTensorV2，为什么不用ToTensor，而是ToTensorV2

DLLXW commented 3 years ago

您好，谢谢您的分享！请问在data-science-competition/kaggle/Cassava Leaf Disease Classification/里，数据增强用的是ToTensorV2，为什么不用ToTensor，而是ToTensorV2

看了官方文档，这里图像分类可以也可以用ToTensor，V2的话会针对mask和image都work,针对分割任务的吧