-
Notifications
You must be signed in to change notification settings - Fork 89
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to create and train the dataset #24
Comments
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 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. |
@mgonzs13 Thanks for your answering, I will try later! |
Hey @cds4488, how is this going? |
Hello, may I ask how to use this model to train the dataset you have created? What should be the format and production of the RGB dataset and the RGBD dataset? Because what I want to do is to create a dataset of transparent objects in the laboratory. Thank you!
The text was updated successfully, but these errors were encountered: