If I want to train on my datasets

sajjad-sh33 / YOLO_SAM2

Self-Prompting Polyp Segmentation in Colonoscopy Using Hybrid YOLO-SAM 2 Model

29 stars 5 forks source link

If I want to train on my datasets #1

Open jzw0707 opened 1 month ago

jzw0707 commented 1 month ago

Hi,nice work. I would like to ask if I want to train on my own data set, whether the format used by the data set is yolo's txt format?

sajjad-sh33 commented 1 month ago

Hi, yes, we use the YOLO’s txt format for training. You can find samples in the Creat_YAML.py file where we convert .csv or ground truth images to .txt files.

jzw0707 commented 4 weeks ago

Thanks！