Add disabling LM filter (#75)

luxonis · Dec 26, 2024 · 03a05cf · 03a05cf
1 parent 931ddcb
commit 03a05cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/examples/generate_dataset_and_train_yolo.ipynb b/examples/generate_dataset_and_train_yolo.ipynb
@@ -59,6 +59,7 @@
     "!datadreamer --save_dir generated_dataset \\\n",
     "             --class_names robot tractor horse car person bear \\\n",
     "             --prompts_number 100 \\\n",
+    "             --disable_lm_filter \\\n",
     "             --prompt_generator simple \\\n",
     "             --num_objects_range 2 3 \\\n",
     "             --image_generator sdxl-turbo \\\n",

diff --git a/examples/helmet_detection.ipynb b/examples/helmet_detection.ipynb
@@ -55,6 +55,7 @@
     "!datadreamer --save_dir gen_dataset_helmet_10000_turbo_tiny \\\n",
     "             --class_names helmet \\\n",
     "             --prompts_number 10000 \\\n",
+    "             --disable_lm_filter \\\n",
     "             --prompt_generator tiny \\\n",
     "             --num_objects_range 1 1 \\\n",
     "             --image_generator sdxl-turbo \\\n",