THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
799 stars 60 forks source link

数据集的形式还有标签应该放在哪? #45

Open yi-coder18 opened 7 months ago

yi-coder18 commented 7 months ago

您好,我现在准备用自己的数据集进行训练,但是readme里面写的,把图片放在imagenet文件夹的train和val文件夹下,我想问一下train和val又是什么样的格式,里面的文件夹名字是类别吗?我不太清楚。标签也放在train和val里面吗,我从代码里面没有看出来标签应该放在哪

yi-coder18 commented 7 months ago

还有,我用的不是Imagenet里面的,而是实验室自己的医学图像的数据集

jameslahm commented 7 months ago

您好,train和val里面的文件夹文字是类别,标签是按照文件夹组织的,会根据train和val下面的所有文件夹名字得到所有类别。

yi-coder18 commented 7 months ago

您好,我还是不太懂标签文件应该放在哪个文件夹里面?还有一个问题就是,这个分类的意思是,训练里面的图都是二分类吗,就是一个主体一个背景吗,但是我的数据集里面的图都是要分割出几种东西的,就是标签是0,1,2,3…这种,那这个怎么分类,类别名怎么办?

---- 回复的原邮件 ---- | 发件人 | Wang @.> | | 发送日期 | 2024年03月25日 17:21 | | 收件人 | THU-MIG/RepViT @.> | | 抄送人 | yi-coder18 @.>, Author @.> | | 主题 | Re: [THU-MIG/RepViT] 数据集的形式还有标签应该放在哪? (Issue #45) |

您好,train和val里面的文件夹文字是类别,标签是按照文件夹组织的,会根据train和val下面的所有文件夹名字得到所有类别。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

jameslahm commented 7 months ago

您好,

  1. 这个分类是二分类,只有一个主体,然后包含相同主体的图片放在一个文件夹下,文件夹名对应其类别
  2. 一张图片分割出几种东西这种,目前这个不支持
18222why commented 7 months ago

还有,我用的不是Imagenet里面的,而是实验室自己的医学图像的数据集 可以的,我这边使用自己数据集运行成功了,按照readme图片格式就行,文件夹下面直接是图片

seany56 commented 6 months ago

还有,我用的不是Imagenet里面的,而是实验室自己的医学图像的数据集 可以的,我这边使用自己数据集运行成功了,按照readme图片格式就行,文件夹下面直接是图片

您好,你是直接将图片设置好后,直接运行的train函数吗