^jz , d dl mZ d dlZd dlZd dlmZmZmZ d dlm Z d dl
m Z d dl m
Z
mZmZ d dlZd dlZd dlm Z d dlmZ d dlmZ d d lmZ d d
lm Z d d l m Z m Z d d l m!Z!m"Z"m#Z#m$Z$ d d
l%m&Z& d dl'm(Z( dZ) G d d Z* G d d Z+ G d d Z,y) ) annotationsN)Counter defaultdictdeque)Iterable) lru_cache)AnyLiteralcast)CLASS_NAME_DATA_FIELD)
Detections)
cross_product)Color)draw_rectangle draw_text)PointPositionRectVector)
overlay_image)SupervisionWarnings
c e Zd ZdZej
ej ej ej fdf d
dZ e
dd Z e
dd Z e
dd Z
e
dd Z ddZedd Z dd
Zdd Zy )LineZonea
This class is responsible for counting the number of objects that cross a
predefined line.
!!! warning
LineZone uses the `tracker_id`. Read
[here](/latest/trackers/) to learn how to plug
tracking into your inference pipeline.
Attributes:
in_count: The number of objects that have crossed the line from outside
to inside.
out_count: The number of objects that have crossed the line from inside
to outside.
in_count_per_class: Number of objects of each class that have
crossed the line from outside to inside, keyed by `class_id`
(`int` for classified detections, `None` for unclassified ones).
out_count_per_class: Number of objects of each class that have
crossed the line from inside to outside, keyed by `class_id`
(`int` for classified detections, `None` for unclassified ones).
Example:
```pycon
>>> import numpy as np
>>> import supervision as sv
>>> def frames_generator():
... yield np.zeros((1080, 1920, 3), dtype=np.uint8)
...
>>> start = sv.Point(x=0, y=100)
>>> end = sv.Point(x=200, y=100)
>>> line_zone = sv.LineZone(start=start, end=end)
>>> for frame in frames_generator():
... detections = sv.Detections(
... xyxy=np.array([[10, 110, 20, 150]]),
... tracker_id=np.array([1])
... )
... crossed_in, crossed_out = line_zone.trigger(
... detections
... )
...
>>> line_zone.in_count, line_zone.out_count
(0, 0)
>>> for frame in frames_generator():
... detections = sv.Detections(
... xyxy=np.array([[10, 50, 20, 90]]),
... tracker_id=np.array([1])
... )
... crossed_in, crossed_out = line_zone.trigger(
... detections
... )
...
>>> line_zone.in_count, line_zone.out_count
(1, 0)
```
c R t || _ j j _ t d|dz _ t
fd _ t _ t _
| _ t j s t d i _ y)a
Args:
start: The starting point of the line.
end: The ending point of the line.
triggering_anchors: A list of positions specifying which anchors of
the detections bounding box to consider when deciding on whether
the detection has passed the line counter or not. By default,
this contains the four corners of the detection's bounding box.
minimum_crossing_threshold: Detection needs to be seen on the other
side of the line for this many frames to be considered as having
crossed the line. This is useful when dealing with unstable
bounding boxes or when detections may linger on the line.
startend)vector r c 0 t j S )N)maxlen)r crossing_history_lengthselfs j/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/detection/line_zone.pyz#LineZone.__init__..x s T-I-I J z#Triggering anchors cannot be empty.N)r r $_calculate_region_of_interest_limitslimitsmaxr$ r crossing_state_historyr _in_count_per_class_out_count_per_classtriggering_anchorslist
ValueErrorclass_id_to_name)r& r r r0 minimum_crossing_thresholds ` r' __init__zLineZone.__init__[ s 2 5c2 ??t{{?S '*1.H1.L'M$ J K
# 9@ 9@!"4 D++, BC C02 r) c H t | j j S N)sumr. valuesr% s r' in_countzLineZone.in_count s 4++22455r) c H t | j j S r7 )r8 r/ r9 r% s r' out_countzLineZone.out_count s 4,,33566r) c , t | j S r7 )dictr. r% s r' in_count_per_classzLineZone.in_count_per_class s D,,--r) c , t | j S r7 )r> r/ r% s r' out_count_per_classz LineZone.out_count_per_class s D--..r) c L t j t | d }t j t | d }t | dk( r||fS |j t j
dt ||fS | j | | j | \ }}}|j t |j ndgt | z }t t ||j D ] \ }\ } }
|| s|| r|| r|| } | j |
| f } | j | t | | j k rZ| d }
| j! |
dkD rt| r | j" | xx dz
cc< d||< | j$ | xx dz
cc< d||< ||fS )a
Update the `in_count` and `out_count` based on the objects that cross the line.
Args:
detections: A Detections object for which to update the counts.
Returns:
A tuple of two boolean NumPy arrays. The first array indicates which
detections have crossed the line from outside to inside. The second
array indicates which detections have crossed the line from inside to
outside.
Fr NzLine zone counting skipped. LineZone requires tracker_id. Refer to https://supervision.roboflow.com/latest/trackers for more information.)categoryr T)npfulllen
tracker_idwarningswarnr _update_class_id_to_name_compute_anchor_sidesclass_idr1 enumeratezipr- appendr$ countr. r/ )r&
detections
crossed_in crossed_out in_limitshas_any_left_triggerhas_any_right_trigger class_idsirL rG
tracker_statecrossing_history oldest_states r' triggerzLineZone.trigger s WWS_e4
ggc*ou5 z?a {* * ( MM -
{* * %%j1
& &z 2 ? ')> "".
$$ %#j/) *3 :00 1*
& %A%* Q<#A&+@+C"6q"9M # : :J;Q R # #M 2#$t'C'CC+A.L %%l3a7 ((2a72 $
1
))(3q83!% A3 &6 ;&&r) c T | j }|dk( r t d | j j | j j z
}| j j
| j j
z
}||z }||z }| }|}t
| j t | j j |z | j j
|z }t
| j t | j j |z
| j j
|z
} || fS )Nr z+The magnitude of the vector cannot be zero.)xyr ) magnituder2 r r^ r r_ r r )
r r` delta_xdelta_y
unit_vector_x
unit_vector_yperpendicular_vector_xperpendicular_vector_ystart_region_limitend_region_limits
r' r* z-LineZone._calculate_region_of_interest_limits s $$ > JK K**,, /**,, / )+
)+
"/ !. #,,,,..#99,,..#99
"****,,!77**,,!77
"#333r) c t | dkD sJ |j J t j | j D cg c] }|j | c} }t
|| j d }t
|| j d }|dkD |dkD k( }t j |d }t
|| j dk }t j |d }t j | d } ||| fS c c}w )u
Find if detections' anchors are within the limit of the line
zone and which anchors are on its left and right side.
Assumes:
* At least 1 detection is provided
* Detections have `tracker_id`
The limit is defined as the region between the two lines,
perpendicular to the line zone, and passing through its start
and end points, as shown below:
Limits:
```
| IN ↑
| |
OUT o---LINE---o OUT
| |
↓ IN |
```
Args:
detections: The detections to check.
Returns:
All 3 arrays are boolean arrays of shape (N, ) where N is the
number of detections. The first array, `in_limits`, indicates
if the detection's anchor is within the line zone limits.
The second array, `has_any_left_trigger`, indicates if the
detection's anchor is on the left side of the line zone.
The third array, `has_any_right_trigger`, indicates if the
detection's anchor is on the right side of the line zone.
r r )axis) rF rG rD arrayr0 get_anchors_coordinatesr r+ allr any)
r& rQ anchor all_anchorscross_products_1cross_products_2rT triggersrU rV s
r' rK z LineZone._compute_anchor_sides s H :"""$$000hh # 5 5
226:
)dkk!nE(dkk!nE &)/?!/CD FF91- dkk:Q> !vvhQ 7 "yq 9 .0EEE#
s C<c f |j j t }||j J |j y|$|j D ci c] }|t | }}n)t |j | D ci c] \ }}||
}}}| j j | yc c}w c c}}w )z
Update the attribute keeping track of which class
IDs correspond to which class names.
Assumes that class_names are only provided when class_ids are.
N)datagetr rL strrN r3 update)r& rQ class_namesrL new_names
class_names r' rJ z!LineZone._update_class_id_to_name3 s !oo))*?@ "j&9&9&EEE & AKATATUX3x=0UIU -0
0C0C[,Q(Hj *$I
$$Y/
Vs
B(<
B-N)r r r r r0 zIterable[Position]r4 int)returnr| )r} zdict[int | None, int])rQ r
r} z3tuple[npt.NDArray[np.bool_], npt.NDArray[np.bool_]])r r r} ztuple[Vector, Vector])rQ r
r} zJtuple[npt.NDArray[np.bool_], npt.NDArray[np.bool_], npt.NDArray[np.bool_]])rQ r
r} None)__name__
__module__ __qualname____doc__r TOP_LEFT TOP_RIGHT BOTTOM_LEFT BOTTOM_RIGHTr5 propertyr: r< r? rA r\ staticmethodr* rK rJ r) r' r r s >J
! ! 2
+,$3$3 $3 / $3 %($3L 6 6 7 7 . . / /E'$E' <E'N 4 4>9F$9F S9Fv0r) r c R e Zd Zdej dej
ddddddddddf ddZ dd Zdd
Z dd Z dd Z
dd
Z e e
d dd Zy)LineZoneAnnotatorr! g ? ?r NTFc || _ || _ || _ || _ || _ || _ || _ |r|nd| _ | r| nd| _ |
| _ | | _
| | _ |
| _ || _
y)a
A class for drawing the `LineZone` and its detected object count
on an image.
Args:
thickness: Line thickness.
color: Line color.
text_thickness: Text thickness.
text_color: Text color.
text_scale: Text scale.
text_offset: How far the text will be from the line.
text_padding: The empty space in the text box, surrounding the text.
custom_in_text: Write something else instead of "in".
custom_out_text: Write something else instead of "out".
display_in_count: Pass `False` to hide the "in" count.
display_out_count: Pass `False` to hide the "out" count.
display_text_box: Pass `False` to hide the text background box.
text_orient_to_line: Match text orientation to the line.
Recommended to set to `True`.
text_centered: Pass `False` to disable text centering. Useful
when the label overlaps something important.
inoutN) thicknesscolortext_thickness
text_color
text_scale text_offset text_paddingin_textout_textdisplay_in_countdisplay_out_countdisplay_text_boxtext_orient_to_line
text_centered)r& r r r r r r r custom_in_textcustom_out_textr r r r r s r' r5 zLineZoneAnnotator.__init__K sw P ( !
#1!+!+"-!-.>@
&&**::< JJ NN[[
//((*[[
//((*[[
\\N"\%:%:$;<mm_B|'='=&>? ! 1 1, ? d++T 2
t--u 5,
'D(K !Q&d.F.F&& , 3 3 : : + ' )) * + * , r) c b |j j j }|j j j }|d |d z
}|d |d z
}|dk( rd}||dk rdz
}|S dz
}|S t j
t j ||z }||dk rdndz
}|S )z
Calculate the line counter angle (in degrees).
Args:
line_zone: The line zone object.
Returns:
Line counter angle, in degrees.
r r g V@ )r r r r mathdegreesatan)r& r start_point end_pointra rb
line_angles r' r z!LineZoneAnnotator._get_line_angle s &&,,<<> $$((88: A,Q /A,Q / a< J 1# 3J
34 3J
dii'0A&BCJ 1#! 3Jr) c ^ | j | }| j rYt |j j |j j
j j }t | }n|j j
j } t | }t t j t j | |dz | j z z }
t t j t j | |dz | j z z } |dxx |
z cc< |dxx | z cc< t t j t j | | j |z z } t t j t j | | j |z z }
|r|dxx | z
cc< |dxx |
z cc< n|dxx | z cc< |dxx |
z
cc< t! |d |dz z
d }t! |d |dz z
d }||fS )a
Calculate insertion anchor in frame to position the center of the count image.
Args:
line_zone: The line counter object used for counting.
text_width: Text width.
text_height: Text height.
is_in_count: Whether the count should be placed over or below line.
label_dimension: Size of the label image. Assumes the
label is rectangular.
Returns:
xy, point in an image where the label will be placed.
r r! r r )r r r r r r r r r1 r| r cosradiansr sinr r, )r& r
text_width text_heightr label_dimensionr mid_pointro r move_along_x move_along_ymove_perpendicular_xmove_perpendicular_yx1y1s r' _calculate_anchor_in_framez,LineZoneAnnotator._calculate_anchor_in_frame s , )))4
&&,,)2B2B2F2Ff__&
)_F!((,,<<>I)_F j12 >D$5$557 L j12 >D$5$557 L
1I %I 1I %I " HHT\\*- .$2B2B[2P Q
# HHT\\*- .$2B2B[2P Q
1I- -I 1I- -I 1I- -I 1I- -I
_ 111
5
_ 111
52v
r) c
t j |t j | j | j d \ }}|r,|xj
t
| j |z z c_ n+|xj
t
| j |z z
c_ t |||| j | j | j | j | j r | j nd |S )a
Draw the count label on the frame. For example: "out: 7".
The label contains horizontal text and is not rotated.
Args:
frame: The entire scene, on which the label will be placed.
line_center: The center of the line zone.
text: The text that will be drawn.
is_in_count: Whether to display the in count (above line)
or out count (below line).
Returns:
The scene with the label drawn on it.
r N)scener text_anchorr r r r background_color)
r getTextSizeFONT_HERSHEY_SIMPLEXr r r_ r| r r r r r r )r& r r r r _r s r' r z#LineZoneAnnotator._draw_basic_label! s * #**DOOT=P=P
; MMS!1!1K!? @ @M MMS!1!1K!? @ @M# .. **+/+@+@TZZd
r) c
| j | }| j || j | j | j | j
| j | j | }|j d |j d k( sJ t j |t j | j | j d \ }}| j |||||j d } t ||| }|S )a
Draw the count label on the frame. For example: "out: 7".
The label is oriented to match the line angle.
Args:
frame: The entire scene, on which the label will be placed.
line_zone: The line zone responsible for counting objects crossing it.
text: The text that will be drawn.
is_in_count: Whether to display the in count (above line)
or out count (below line).
Returns:
The scene with the label drawn on it.
)r r r r
text_box_showtext_box_colorr r r )r r r r r )r _make_label_imager r r r r r shaper r r r r )
r& r r r r r label_imager r label_anchors
r' r z&LineZoneAnnotator._draw_oriented_labelL s , " 1 1) <,, .. ** // :: 1 -
#{'8'8';;;;"%// #**DOOT=P=P#
#
K 66 !## ' - -a 0 7
e[,? r) )maxsizec t j | t j || d \ }} t t || |dz z dz }
|
|
f} t |
dz |
dz } t
j g | dt j }
t
j g | dt j }t | | ||| }t d|
||r|ndd | t d|t j |rt j ndd | t
j |
|f }
d
|d z cxk rd k r7n n4t j |
d
j t j }
| }t j | j# |d }t j$ |
|| }
t' t( j* t j |
S )aY
Create the small text box displaying line zone count. E.g. "out: 7".
Args:
text: The text to display.
text_scale: The scale of the text.
text_thickness: The thickness of the text.
text_padding: The padding around the text.
text_color: The color of the text.
text_box_show: Whether to display the text box.
text_box_color: The color of the text box.
line_angle_degrees: The angle of the line in degrees.
Returns:
The label of shape (H, W, 4), in BGRA format.
r r! r )dtyper )r r r r r N)r r r Z ih i r )flipCode)scaler )r r r r| r, r rD zerosuint8r> r r WHITEdstackflipastypegetRotationMatrix2Das_xy_float_tuple
warpAffiner nptNDArray)r r r r r r r r r r annotation_dimannotation_shapeannotation_center
annotationannotation_alpha text_argsrotation_anglerotation_matrixs r' r z#LineZoneAnnotator._make_label_image s : #&// #**J#
#
K c*k :\A=M MQT TU *N; !.A"5~7J KXX 4 0 4! 4BHHE
88$:&6$:$:"((K$()!)% %
!/<^$
" {{,9U[[t
YY
,< =>
"S( .3 . *r:AA"((KJ,, 11 / / 1>
^^JAQR
CKK):66r) ) r r| r r r r| r r r floatr r r r| r
str | Noner r r boolr r r r r r r r )r npt.NDArray[np.uint8]r r r} r )r r r} r ) r r r r| r r| r r r r| r} ztuple[int, int])
r r r r r rw r r r} r )
r r r r r rw r r r} r )r rw r r r r| r r| r r r r r r r r r} r )r r r r r BLACKr5 r r r r r r r r r r) r' r r J s {{ !KK %)&*!%"&!%$) " 5151 51 51
51
51 51 51 #51 $51 51 51 51 " 51 51nD*D:BD DL2>> > >
>
>
>@)$) ) )
)
)V1$1 1 1
1
1f rE7E7 E7 E7
E7
E7 E7 E7 "E7
E7 E7r) r c
e Zd Zej ej dddej dddd d
dZ d d d Z y)
LineZoneAnnotatorMulticlassr i g ?r F) table_position table_color table_margin
table_paddingtable_max_widthr r r force_draw_class_idsc |t j t j t j t j hvr t d || _ || _ || _ || _ || _
|| _ || _ || _
| | _ y)a
Draw a table showing how many items of each class crossed each line.
Args:
table_position: The position of the table.
table_color: The color of the table.
table_margin: The margin of the table from the image border.
table_padding: The padding of the table.
table_max_width: The maximum width of the table.
text_color: The color of the text.
text_scale: The scale of the text.
text_thickness: The thickness of the text.
force_draw_class_ids: Instead of writing the class names,
on the table, write the class IDs. E.g. instead of `person: 6`,
write `0: 6`.
z]Invalid table position. Supported values are: TOP_LEFT, TOP_RIGHT, BOTTOM_LEFT, BOTTOM_RIGHT.N)r r r r r r2 r r r r r r r r r )
r& r r r r r r r r r s
r' r5 z$LineZoneAnnotatorMulticlass.__init__ s D ! ! "
C
-& ( * . $$ ,$8!r) Nc |(t t | D cg c] }d|dz d
}}t | t | k7 r t d dg}t || D ] \ }}|j | |j
}d|j fd|j ffD ] \ } }
|
s |j d| d |
j D ]T \ } } | j r t | }
n | d }
n|j | t | }
|j d
|
d | V d \ }}|D ]\ }t j |t j | j | j d
\ }}|t" z
}t% || }||z
}^ |d| j& z z
}|d| j& z z
}|j( d
d| j* z z
}t- || }t- || j. }t0 j2 | j* | j* ft0 j4 |j( d |z
| j* z
| j* ft0 j6 | j* |j( d
|z
| j* z
ft0 j8 |j( d |z
| j* z
|j( d
|z
| j* z
fi}|| j: \ }}t= |||| }t? ||| j@ d }tC | D ] \ }}t j |t j | j | j d
\ }}|t" z
}|| j& z }|| j& z |dz |z z }t jD ||||ft j | j | jF jI | j t jJ |S c c}w )a
Draws a table with the number of objects of each class that crossed each line.
Args:
frame: The image on which the table will be drawn.
line_zones: The line zones to be annotated.
line_zone_labels: The labels, one for each line zone. If not
provided, the default labels will be used.
Returns:
The image with the table drawn on it.
zLine r :z5The number of line zones and their labels must match.zLine Crossings:InOut r~ z r )r r r r! )r^ r_ widthheightr )r rectr r )imgr orgfontFace fontScaler r r )&rangerF r2 rN rO r3 r? rA itemsr rw rv r r r r r TEXT_MARGINr, r r r minr r r r r r r r r r rM putTextr r r )r& r
line_zonesline_zone_labelsrX
text_linesr line_zone_labelr3 directioncount_per_classrL rP r{ table_width table_heightr r r table_max_height position_maptable_x1table_y1
table_rectr anchor_xanchor_ys r' r z$LineZoneAnnotatorMulticlass.annotate s" & #:?J:P QQ%Awa 0 Q Q z?c"2 3 3 TU U'(
*-j:J*K B &I o . ( 9 9 y334 556/
B* ? ' !!Ai["23'6'<'<'> BOHe00%(]
!)%+
%5%9%9(CM%R
%%:,b&@AB
B B* %)! \ (D&)ooc..ATAT'' #J
; &K k:6K K 'L
( q4---- D . ... ;;q>A0A0A,AA<)9: +t';';<
1 143D3D E A,t/@/@@!!!
!! A -0A0AA#
! ! A,t/@/@@ A -0A0AA$
*$*=*= >((+l
j0@0@B
!, GAt __c..ATAT NA{
; &K $"4"44H $"4"44A7LLH KK x( 11 //oo,,. --
& i Rs O;)r z[Literal[Position.TOP_LEFT, Position.TOP_RIGHT, Position.BOTTOM_LEFT, Position.BOTTOM_RIGHT]r r r r| r r| r r| r r r r r r| r r r7 )r r r zlist[LineZone]r zlist[str] | Noner} r )
r r r r r r r r r5 r r r) r' r r s
"[[ " !KK %*!59
59 59 59 59 59 59 59 59 #!59v .2 h$h #h + h
hr) r )-
__future__r r rH collectionsr r r collections.abcr functoolsr typingr r
r r numpyrD numpy.typingr supervision.configr supervision.detection.corer
$supervision.detection.utils.internalr supervision.draw.colorr supervision.draw.utilsr r supervision.geometry.corer r r r supervision.utils.imager supervision.utils.internalr r r r r r r) r' r9 so " 3 3 $ % %
4 1 > ( < C C 1 : m0 m0` |7 |7~ ` `r)