o
    à¥µiŒ  ã                   @   sT   d dl mZ d dlmZmZmZ d dlZd dlm	Z	 ed Z
eG dd„ de	ƒƒZdS )é    )Ú	dataclass)ÚOptionalÚTupleÚUnionN)ÚModelOutputBase)ztorch.Tensorz	tf.Tensorc                   @   sB   e Zd ZU dZdZeed< dZeed< dZeed< dZ	eed< dS )ÚDetectionOutputaæ  The output class for object detection models.

    Args:
        class_ids (`Tensor`, *optional*): class id for each object.
        boxes (`Tensor`, *optional*): Bounding box for each detected object in  [left, top, right, bottom] format.
        scores (`Tensor`, *optional*): Detection score for each object.
        keypoints (`Tensor`, *optional*): Keypoints for each object using four corner points in a 8-dim tensor
            in the order of (x, y) for each corner point.

    NÚ	class_idsÚscoresÚboxesÚ	keypoints)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚTensorÚ__annotations__r	   r
   r   © r   r   úQ/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/outputs/cv_outputs.pyr      s   
 r   )Údataclassesr   Útypingr   r   r   ÚnumpyÚnpÚmodelscope.outputs.outputsr   r   r   r   r   r   r   Ú<module>   s   