diff --git a/README.md b/README.md index 17921df..45ef010 100644 --- a/README.md +++ b/README.md @@ -32,6 +32,24 @@ datadreamer --class_names person moon robot - **Efficient and Potent Models**: The primary objective of `DataDreamer` is to enable the creation of compact models that are both size-efficient for integration into any device and robust in performance for specialized tasks. +### ✨ New: Pre-annotate Real Data with DataDreamer + +`DataDreamer` helps you accelerate your annotation process by pre-annotating real data with minimal effort. Simply provide your dataset, and `DataDreamer` generates high-quality initial annotations for further refinement. + +Available tasks: classification, object detection, and instance segmentation. + + + +#### Example + +Run the following to pre-annotate images in your dataset: + +```bash +datadreamer --task instance-segmentation --image_annotator owlv2-slimsam --save_dir dataset_path --class_names dumpling --annotate_only +``` + +📚 **Tutorial**: [Training a Semantic Segmentation Model using luxonis-train and DataDreamer](https://github.com/luxonis/depthai-ml-training/blob/main/training/luxonis-train/train_semantic_segmentation_model_datadreamer.ipynb) + ## 📜 Table of contents - [🚀 Quickstart](#quickstart) @@ -157,7 +175,7 @@ datadreamer --config ### 🔧 Additional Parameters -- `--task`: Choose between detection, classification and instance segmentation. Default is `detection`. +- `--task`: Choose between `detection`, `classification` and `instance-segmentation`. Default is `detection`. - `--dataset_format`: Format of the dataset. Defaults to `raw`. Supported values: `raw`, `yolo`, `coco`, `luxonis-dataset`, `cls-single`. - `--split_ratios`: Split ratios for train, validation, and test sets. Defaults to `[0.8, 0.1, 0.1]`. - `--num_objects_range`: Range of objects in a prompt. Default is 1 to 3. diff --git a/images/dumplings_seg_preannotation.gif b/images/dumplings_seg_preannotation.gif new file mode 100644 index 0000000..f197bcb Binary files /dev/null and b/images/dumplings_seg_preannotation.gif differ