fix: fix example notebook

luxonis · Dec 11, 2023 · 0b981a0 · 0b981a0
1 parent 916647b
commit 0b981a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/generate_dataset_and_train_yolo.ipynb b/examples/generate_dataset_and_train_yolo.ipynb
@@ -83,7 +83,8 @@
         "- `--use_tta`: Toggle test time augmentation for object detection. Default is True.\n",
         "- `--enhance_class_names`: Enhance class names with synonyms. Default is False.\n",
         "- `--use_image_tester`: Use image tester for image generation. Default is False. Image tester is a model that checks if the generated image contains the desired object.\n",
-        "- `--image_tester_patience`: Patience level for image tester. Default is 1. Number of images to run through the image tester before giving up.\n"
+        "- `--image_tester_patience`: Patience level for image tester. Default is 1. Number of images to run through the image tester before giving up.\n",
+        "- `--seed`: Set a random seed for image and prompt generation. Default is 42.\n"
       ]
     },
     {