Referentie voor ultralytics/solutions/distance_calculation.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/distance_calculation .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.solutions.distance_calculation.DistanceCalculation
Een klasse om de afstand tussen twee objecten in een real-time videostream te berekenen op basis van hun sporen.
Broncode in ultralytics/solutions/distance_calculation.py
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 |
|
__init__()
Initialiseert de afstandberekeningsklasse met standaardwaarden voor Visual, Image, track en afstand parameters.
Broncode in ultralytics/solutions/distance_calculation.py
calculate_centroid(box)
Bereken het middelpunt van de bounding box.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
box |
list
|
Gegevens voor de begrenzende box |
vereist |
calculate_distance(centroid1, centroid2)
Bereken de afstand tussen twee middelpunten.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
centroid1 |
point
|
Eerste bounding box gegevens |
vereist |
centroid2 |
point
|
Gegevens tweede bounding box |
vereist |
Broncode in ultralytics/solutions/distance_calculation.py
display_frames()
Weergave frame.
Broncode in ultralytics/solutions/distance_calculation.py
extract_tracks(tracks)
Haalt resultaten uit de verstrekte gegevens.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
tracks |
list
|
Lijst met sporen verkregen uit het objectvolgproces. |
vereist |
Broncode in ultralytics/solutions/distance_calculation.py
mouse_event_for_distance(event, x, y, flags, param)
Deze functie is ontworpen om regio's te verplaatsen met muisgebeurtenissen in een real-time videostream.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
event |
int
|
Het type muisgebeurtenis (bijvoorbeeld cv2.EVENT_MOUSEMOVE, cv2.EVENT_LBUTTONDOWN, etc.). |
vereist |
x |
int
|
De x-coördinaat van de muisaanwijzer. |
vereist |
y |
int
|
De y-coördinaat van de muisaanwijzer. |
vereist |
flags |
int
|
Eventuele vlaggen die bij de gebeurtenis horen (bijv. cv2.EVENT_FLAG_CTRLKEY, cv2.EVENT_FLAG_SHIFTKEY, enz.). |
vereist |
param |
dict
|
Extra parameters die je aan de functie wilt doorgeven. |
vereist |
Broncode in ultralytics/solutions/distance_calculation.py
set_args(names, pixels_per_meter=10, view_img=False, line_thickness=2, line_color=(255, 255, 0), centroid_color=(255, 0, 255))
Configureert de afstandsberekening en weergaveparameters.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
names |
dict
|
objectdetectieklassen namen |
vereist |
pixels_per_meter |
int
|
Aantal pixels in meter |
10
|
view_img |
bool
|
Vlag die frameweergave aangeeft |
False
|
line_thickness |
int
|
Lijndikte voor begrenzende vakken. |
2
|
line_color |
RGB
|
kleur van hartlijn |
(255, 255, 0)
|
centroid_color |
RGB
|
kleuren van bbox-centroïden |
(255, 0, 255)
|
Broncode in ultralytics/solutions/distance_calculation.py
start_process(im0, tracks)
Bereken de afstand tussen twee bounding boxes op basis van trackinggegevens.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
im0 |
nd array
|
Afbeelding |
vereist |
tracks |
list
|
Lijst met sporen verkregen uit het objectvolgproces. |
vereist |
Broncode in ultralytics/solutions/distance_calculation.py
Gemaakt op 2024-01-05, Bijgewerkt op 2024-05-08
Auteurs: Burhan-Q (1), AyushExel (1), RizwanMunawar (1)