search

sajjad-sh33 / YOLO_SAM2

Self-Prompting Polyp Segmentation in Colonoscopy Using Hybrid YOLO-SAM 2 Model
40 stars 5 forks source link

If I want to train on my datasets #1

Open jzw0707 opened 2 months ago

jzw0707 commented 2 months 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 2 months 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 2 months ago

Thanks！