Skip to content

Commit a326fe5

Browse files
committed
fix of library docs
1 parent b8d4a88 commit a326fe5

3 files changed

Lines changed: 31 additions & 24 deletions

File tree

examples/example_extra_functions.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,11 @@
8888
" show_confidences (bool): If True and show_class=True, confidences near class are visualized.\n",
8989
" axis_off (bool): If True, axis is turned off in the final visualization.\n",
9090
" show_classes_list (list): If empty, visualize all classes. Otherwise, visualize only classes in the list.\n",
91+
" inference_extra_args (dict): Dictionary with extra ultralytics inference parameters\n",
9192
" return_image_array (bool): If True, the function returns the image (BGR np.array) instead of displaying it.\n",
9293
"\n",
9394
" Returns:\n",
94-
" None"
95+
" None or np.array if return_image_array==True"
9596
]
9697
},
9798
{

examples/example_patch_based_inference.ipynb

Lines changed: 26 additions & 23 deletions
Large diffs are not rendered by default.

patched_yolo_infer/nodes/MakeCropsDetectThem.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,9 @@ class MakeCropsDetectThem:
3030
model: Pre-initialized model object. If provided, the model will be used directly
3131
instead of loading from model_path.
3232
memory_optimize (bool): Memory optimization option for segmentation (less accurate results)
33+
batch_inference (bool): Batch inference of image crops through a neural network instead of
34+
sequential passes of crops (ps: Faster inference, higher memory use)
35+
inference_extra_args (dict): Dictionary with extra ultralytics inference parameters
3336
3437
Attributes:
3538
model: YOLOv8 model loaded from the specified path.

0 commit comments

Comments
 (0)