How to create and train the dataset about yolov8_ros HOT 3 CLOSED

Hey @cds4488, to use your custom dataset, you have to train YOLOv8 as normal. Then, pass the new model path to the launch file using the model parameter.

For instance, you have the following example. Replace the coco128.yaml with you YAML file and choose the base YOLOv8 model.

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.yaml')  # build a new model from YAML
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

# Train the model
results = model.train(data='coco128.yaml', epochs=100, imgsz=640)

Btw, there is no RGBD dataset, I am using the depth images along with the bounding boxes to create the 3D boxes.

from yolov8_ros.

@mgonzs13 Thanks for your answering, I will try later!

from yolov8_ros.

Hey @cds4488, how is this going?

from yolov8_ros.