Referentie voor ultralytics/solutions/speed_estimation.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/speed_estimation .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.solutions.speed_estimation.SpeedEstimator
Een klasse om de snelheid van objecten in een real-time videostream te schatten op basis van hun sporen.
Broncode in ultralytics/solutions/speed_estimation.py
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 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
|
__init__()
Initialiseert de klasse Snelheidsschatter met standaardwaarden voor de parameters Visueel, Beeld, Spoor en Snelheid.
Broncode in ultralytics/solutions/speed_estimation.py
calculate_speed(trk_id, track)
Berekening van de snelheid van het object.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
trk_id |
int
|
object track id. |
vereist |
track |
list
|
traceergeschiedenis voor het tekenen van paden |
vereist |
Broncode in ultralytics/solutions/speed_estimation.py
display_frames()
estimate_speed(im0, tracks, region_color=(255, 0, 0))
Bereken object op basis van trackinggegevens.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
im0 |
nd array
|
Afbeelding |
vereist |
tracks |
list
|
Lijst met sporen verkregen uit het objectvolgproces. |
vereist |
region_color |
tuple
|
Kleur om te gebruiken bij het tekenen van gebieden. |
(255, 0, 0)
|
Broncode in ultralytics/solutions/speed_estimation.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/speed_estimation.py
plot_box_and_track(track_id, box, cls, track)
Plot spoor en begrenzingskader.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
track_id |
int
|
object track id. |
vereist |
box |
list
|
object begrenzende doos gegevens |
vereist |
cls |
str
|
objectklasse naam |
vereist |
track |
list
|
traceergeschiedenis voor het tekenen van paden |
vereist |
Broncode in ultralytics/solutions/speed_estimation.py
set_args(reg_pts, names, view_img=False, line_thickness=2, region_thickness=5, spdl_dist_thresh=10)
Configureert de parameters voor snelheidsschatting en -weergave.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
reg_pts |
list
|
Oorspronkelijke lijst met punten die het snelheidsberekeningsgebied definiëren. |
vereist |
names |
dict
|
objectdetectieklassen namen |
vereist |
view_img |
bool
|
Vlag die frameweergave aangeeft |
False
|
line_thickness |
int
|
Lijndikte voor begrenzende vakken. |
2
|
region_thickness |
int
|
Regio dikte voor snelheidsschatting |
5
|
spdl_dist_thresh |
int
|
Euclidische afstandsdrempel voor snelheidslijn |
10
|
Broncode in ultralytics/solutions/speed_estimation.py
store_track_info(track_id, box)
Trackgegevens opslaan.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
track_id |
int
|
object track id. |
vereist |
box |
list
|
object begrenzende doos gegevens |
vereist |
Broncode in ultralytics/solutions/speed_estimation.py
Gemaakt op 2024-01-05, Bijgewerkt op 2024-05-08
Auteurs: Burhan-Q (1), AyushExel (1), RizwanMunawar (1)