sajjad-sh33 / YOLO_SAM2

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

Again, it's a training issue. #2

Open CodeCarrierZ opened 2 months ago

CodeCarrierZ commented 2 months ago

I'm glad to find such high quality papers and code. Since I am doing deep learning modeling, I would like to train the model on another dataset, what code can I use please, because I found that the traditional deep learning training train.py file does not exist here

sajjad-sh33 commented 2 months ago

Hi. There is no train.py file here since we use the Ultralytics training procedure to train. Specifically, the following code does the training:

yolo task=detect mode=train model=yolov8l.pt imgsz=640 data=./YAML/Kvasir.yaml epochs=50 batch=0.90 name=Kvasir

Please note that before running this code, you need to clone the Ultralytics repository and create the dataset format as required. You can find more information in the Ultralytics documentation or in the following YouTube video.