ObjectDetection3DLogger Class
ml_debugger.monitoring.object_detection_3d.object_detection_3d_torchlogger.ObjectDetection3DTorchLogger
Bases: MonitoringLogger, ObjectDetection3DTorchTracer
Base MonitoringLogger for 3D object detection tasks using PyTorch models.
Subclasses must implement the following abstract methods
- detect(model): Static method to check if this subclass handles the model.
- _get_n_class(model): Extract number of classes from the model.
- _register_model_hooks(model): Register forward hooks for logit capture.
- _parse_model_output(model_output): Convert model output to a standardized Detection3DOutput.
Optionally override
- _is_nms_free(): Return True if the model uses NMS-free inference.
__init__(model, model_name, version_name, result_name=None, n_epoch='latest', target_layers=None, additional_fields=None, auto_sync=True, force_table_recreate=False, api_endpoint=None, api_key=None, n_class=None, score_thresh=None, iou_thresh=None, max_detections_per_frame=None)
Initialize 3D object detection monitoring logger.
引数:
| 名前 | タイプ | デスクリプション | デフォルト |
|---|---|---|---|
|
Module
|
PyTorch model to trace. |
必須 |
|
str
|
Name of the ML model. |
必須 |
|
str
|
Version identifier for the ML model. |
必須 |
|
Optional[str]
|
The name of the existing evaluation result to retrieve. |
None
|
|
Union[str, Optional[int]]
|
Filter option for n_epoch value. |
'latest'
|
|
Optional[Dict[str, str]]
|
Mapping of layer aliases to module paths. |
None
|
|
Optional[List[dict]]
|
Extra fields for database schema. |
None
|
|
bool
|
Enable background syncing of logged data. |
True
|
|
bool
|
Whether to drop and recreate existing tables. |
False
|
|
Optional[str]
|
URL of the service API for data upload. |
None
|
|
Optional[str]
|
API key for authenticating with the service. |
None
|
|
Optional[int]
|
Number of classes. Auto-detected from model if None. |
None
|
|
Optional[float]
|
Minimum score threshold for user-visible NMS decisions. |
None
|
|
Optional[float]
|
IoU threshold for user-visible NMS decisions. |
None
|
|
Optional[int]
|
Maximum detections per frame for user-visible output. |
None
|
__call__(model_input, input_ids=None, dataset_type='deploying', **kwargs)
Invoke the logger on a single inference, recording I/O data.
引数:
| 名前 | タイプ | デスクリプション | デフォルト |
|---|---|---|---|
|
Any
|
Input data for the model inference. |
必須 |
|
Optional[List[str]]
|
Identifiers of each input data. |
None
|
|
str
|
Identifier of input dataset. (e.g. 'train', 'val', 'test', 'deploying') |
'deploying'
|
|
Any
|
Additional keyword arguments for parsing and saving I/O data. |
{}
|
戻り値:
| タイプ | デスクリプション |
|---|---|
Any
|
The raw output produced by the model. |
get_hooked_features(layer_name)
Retrieve the captured output for a given layer alias.
引数:
| 名前 | タイプ | デスクリプション | デフォルト |
|---|---|---|---|
|
str
|
Alias of the layer whose activation was captured. |
必須 |
戻り値:
| 名前 | タイプ | デスクリプション |
|---|---|---|
Any |
Any
|
Activation data stored for the specified layer. |
発生:
| タイプ | デスクリプション |
|---|---|
KeyError
|
If no activation has been captured for |
export(output_path=None)
Export extracted features into a ZIP archive.
Uses the internal n_epoch resolved during validator setup to
filter records, consistent with upload() and wait_for_save().
引数:
| 名前 | タイプ | デスクリプション | デフォルト |
|---|---|---|---|
|
Optional[str]
|
Path or directory for saving the ZIP file. If no .zip extension, the default filename is appended. Defaults to cwd. |
None
|
戻り値:
| タイプ | デスクリプション |
|---|---|
Optional[Path]
|
Path to the created ZIP file, or None on non-primary distributed ranks. |