Skip to content

Commit c0148e9

Browse files
committed
docs update
1 parent 2dfa4a6 commit c0148e9

2 files changed

Lines changed: 6 additions & 3 deletions

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ Class implementing cropping and passing crops through a neural network for detec
128128
| inference_extra_args | dict | None | Dictionary with extra ultralytics [inference parameters](https://docs.ultralytics.com/modes/predict/#inference-arguments) (possible keys: half, device, max_det, augment, agnostic_nms and retina_masks) |
129129
| batch_inference | bool | False | Batch inference of image crops through a neural network instead of sequential passes of crops (ps: faster inference, higher gpu memory use). |
130130
| show_processing_status | bool | False | Whether to show the processing status using tqdm. |
131-
| progress_callback | Callable | None | Optional custom callback function, (task: str, current: int, total: int). |
131+
| progress_callback | function | None | Optional custom callback function, (task: str, current: int, total: int). |
132132

133133

134134
**CombineDetections**

patched_yolo_infer/README.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,11 @@ Class implementing cropping and passing crops through a neural network for detec
100100
- **show_crops** (*bool*): Whether to visualize the cropping.
101101
- **resize_initial_size** (*bool*): Whether to resize the results to the original image size (ps: slow operation).
102102
- **memory_optimize** (*bool*): Memory optimization option for segmentation (less accurate results when enabled).
103-
- **inference_extra_args** (*dict*): Dictionary with extra ultralytics [inference parameters](https://docs.ultralytics.com/modes/predict/#inference-arguments) (possible keys: half, device, max_det, augment, agnostic_nms and retina_masks)
104-
- **batch_inference** (*bool*): Batch inference of image crops through a neural network instead of sequential passes of crops (ps: faster inference, higher gpu memory use)
103+
- **inference_extra_args** (*dict*): Dictionary with extra ultralytics [inference parameters](https://docs.ultralytics.com/modes/predict/#inference-arguments) (possible keys: half, device, max_det, augment, agnostic_nms and retina_masks).
104+
- **batch_inference** (*bool*): Batch inference of image crops through a neural network instead of sequential passes of crops (ps: faster inference, higher gpu memory use).
105+
- **show_processing_status** (*bool*): Whether to show the processing status using tqdm. (Default is False)
106+
- **progress_callback** (*function*): Optional custom callback function, (task: str, current: int, total: int).
107+
105108

106109
**CombineDetections**
107110
Class implementing combining masks/boxes from multiple crops + NMS (Non-Maximum Suppression).

0 commit comments

Comments
 (0)