小编rts*_*rts的帖子

在pytorch中加载csv和图像数据集

我正在使用 PyTorch 进行图像分类。我有一个单独的 Images 文件夹,并使用图像 ids 和 labels 训练和测试 csv 文件。我不知道如何组合这些图像和 ID 并转换为张量。

\n
    \n
  1. train.csv :包含图像的所有 ID,如 4325.jpg、2345.jpg、\xe2\x80\xa6so 以及包含猫、狗等标签。
  2. \n
  3. Image_data :包含具有 ID 名称的所有图像。
  4. \n
\n

image machine-learning computer-vision deep-learning pytorch

3
推荐指数
1
解决办法
7531
查看次数