search

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
730 stars 55 forks source link

train my own dataset #12

Closed Jubbuuliiing closed 1 year ago

Jubbuuliiing commented 1 year ago

I want to train my own dataset, how can I modify my class-names?

jameslahm commented 1 year ago

Thanks. If you arrange your dataset like ImageNet, you can just set --data-path to your own dataset.

/path/to/imagenet-1k/
  train/
    class1/
      img1.jpeg
    class2/
      img2.jpeg
  val/
    class1/
      img3.jpeg
    class2/
      img4.jpeg