参考资料 ultralytics/solutions/distance_calculation.py
备注
该文件可在https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/solutions/distance_calculation .py 下找到。如果您发现问题,请通过提交 Pull Request🛠️ 帮助修复。谢谢🙏!
ultralytics.solutions.distance_calculation.DistanceCalculation
根据实时视频流中两个物体的轨迹计算它们之间距离的类。
源代码 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__()
使用视觉、图像、轨迹和距离参数的默认值初始化距离计算类。 参数的默认值。
源代码 ultralytics/solutions/distance_calculation.py
calculate_centroid(box)
计算包围盒的中心点。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
box |
list
|
边界框数据 |
所需 |
calculate_distance(centroid1, centroid2)
计算两个中心点之间的距离。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
centroid1 |
point
|
第一个边界框数据 |
所需 |
centroid2 |
point
|
第二边界框数据 |
所需 |
源代码 ultralytics/solutions/distance_calculation.py
display_frames()
显示框。
源代码 ultralytics/solutions/distance_calculation.py
extract_tracks(tracks)
从提供的数据中提取结果。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
tracks |
list
|
物体追踪过程中获得的轨迹列表。 |
所需 |
源代码 ultralytics/solutions/distance_calculation.py
mouse_event_for_distance(event, x, y, flags, param)
该函数用于在实时视频流中通过鼠标事件移动区域。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
event |
int
|
鼠标事件的类型(如 cv2.EVENT_MOUSEMOVE、cv2.EVENT_LBUTTONDOWN 等)。 |
所需 |
x |
int
|
鼠标指针的 x 坐标。 |
所需 |
y |
int
|
鼠标指针的 Y 坐标。 |
所需 |
flags |
int
|
与事件相关的任何标记(例如,cv2.EVENT_FLAG_CTRLKEY、 cv2.EVENT_FLAG_SHIFTKEY 等)。 |
所需 |
param |
dict
|
您可能希望传递给函数的其他参数。 |
所需 |
源代码 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))
配置距离计算和显示参数。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
names |
dict
|
对象检测类名称 |
所需 |
pixels_per_meter |
int
|
米内像素数 |
10
|
view_img |
bool
|
显示帧的标志 |
False
|
line_thickness |
int
|
边界框的线条粗细 |
2
|
line_color |
RGB
|
中心线的颜色 |
(255, 255, 0)
|
centroid_color |
RGB
|
方框中心点的颜色 |
(255, 0, 255)
|
源代码 ultralytics/solutions/distance_calculation.py
start_process(im0, tracks)
根据跟踪数据计算两个边界框之间的距离。
参数
名称 | 类型 | 说明 | 默认值 |
---|---|---|---|
im0 |
nd array
|
图片 |
所需 |
tracks |
list
|
物体追踪过程中获得的轨迹列表。 |
所需 |
源代码 ultralytics/solutions/distance_calculation.py
创建于 2024-01-05,更新于 2024-05-08
作者:Burhan-Q (1), AyushExel (1), RizwanMunawar (1)Burhan-Q(1),AyushExel(1),RizwanMunawar(1)