Referência para ultralytics/solutions/distance_calculation.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/solutions/distance_calculation.py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.solutions.distance_calculation.DistanceCalculation
Uma classe para calcular a distância entre dois objectos num fluxo de vídeo em tempo real com base nas suas trajectórias.
Código fonte em 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__()
Inicializa a classe de cálculo de distâncias com valores predefinidos para os parâmetros Visual, Imagem, trajeto e distância e distância.
Código fonte em ultralytics/solutions/distance_calculation.py
calculate_centroid(box)
Calcula o centroide da caixa delimitadora.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
box |
list
|
Dados da caixa delimitadora |
necessário |
calculate_distance(centroid1, centroid2)
Calcula a distância entre dois centróides.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
centroid1 |
point
|
Dados da primeira caixa delimitadora |
necessário |
centroid2 |
point
|
Dados da segunda caixa delimitadora |
necessário |
Código fonte em ultralytics/solutions/distance_calculation.py
display_frames()
Apresenta o quadro.
Código fonte em ultralytics/solutions/distance_calculation.py
extract_tracks(tracks)
Extrai resultados dos dados fornecidos.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
tracks |
list
|
Lista de trajectos obtidos a partir do processo de seguimento de objectos. |
necessário |
Código fonte em ultralytics/solutions/distance_calculation.py
mouse_event_for_distance(event, x, y, flags, param)
Esta função foi concebida para mover a região com eventos do rato num fluxo de vídeo em tempo real.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
event |
int
|
O tipo de evento do rato (por exemplo, cv2.EVENT_MOUSEMOVE, cv2.EVENT_LBUTTONDOWN, etc.). |
necessário |
x |
int
|
A coordenada x do ponteiro do rato. |
necessário |
y |
int
|
A coordenada y do ponteiro do rato. |
necessário |
flags |
int
|
Quaisquer sinalizadores associados ao evento (por exemplo, cv2.EVENT_FLAG_CTRLKEY, cv2.EVENT_FLAG_SHIFTKEY, etc.). |
necessário |
param |
dict
|
Parâmetros adicionais que podes querer passar para a função. |
necessário |
Código fonte em 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))
Configura o cálculo da distância e os parâmetros de visualização.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
names |
dict
|
nomes das classes de deteção de objectos |
necessário |
pixels_per_meter |
int
|
Número de pixéis em metros |
10
|
view_img |
bool
|
Bandeira que indica a visualização do quadro |
False
|
line_thickness |
int
|
Espessura da linha para caixas delimitadoras. |
2
|
line_color |
RGB
|
cor da linha dos centróides |
(255, 255, 0)
|
centroid_color |
RGB
|
cores dos centróides da caixa b |
(255, 0, 255)
|
Código fonte em ultralytics/solutions/distance_calculation.py
start_process(im0, tracks)
Calcula a distância entre duas caixas delimitadoras com base em dados de rastreio.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
im0 |
nd array
|
Imagem |
necessário |
tracks |
list
|
Lista de trajectos obtidos a partir do processo de seguimento de objectos. |
necessário |
Código fonte em ultralytics/solutions/distance_calculation.py
Criado em 2024-01-05, Atualizado em 2024-05-08
Autores: Burhan-Q (1), AyushExel (1), RizwanMunawar (1)