рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/engine/validator.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/рдЗрдВрдЬрди/validator.pyред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.engine.validator.BaseValidator
BaseValidator.
рд╕рддреНрдпрд╛рдкрдирдХрд░реНрддрд╛ рдмрдирд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЖрдзрд╛рд░ рд╡рд░реНрдЧред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
args |
SimpleNamespace
|
рд╕рддреНрдпрд╛рдкрдирдХрд░реНрддрд╛ рдХреЗ рд▓рд┐рдП рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рдиред |
dataloader |
DataLoader
|
рд╕рддреНрдпрд╛рдкрди рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбреЗрдЯрд╛рд▓реЛрдбрд░ред |
pbar |
tqdm
|
рд╕рддреНрдпрд╛рдкрди рдХреЗ рджреМрд░рд╛рди рдЕрджреНрдпрддрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдкреНрд░рдЧрддрд┐ рдкрдЯреНрдЯреАред |
model |
Module
|
рдорд╛рдиреНрдп рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореЙрдбрд▓ред |
data |
dict
|
рдбреЗрдЯрд╛ рд╢рдмреНрджрдХреЛрд╢ред |
device |
device
|
рд╕рддреНрдпрд╛рдкрди рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рд╡рд╛рдЗрд╕ред |
batch_i |
int
|
рд╡рд░реНрддрдорд╛рди рдмреИрдЪ рд╕реВрдЪрдХрд╛рдВрдХред |
training |
bool
|
рдХреНрдпрд╛ рдореЙрдбрд▓ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдореЛрдб рдореЗрдВ рд╣реИред |
names |
dict
|
рдХрдХреНрд╖рд╛ рдХреЗ рдирд╛рдоред |
seen |
рд╕рддреНрдпрд╛рдкрди рдХреЗ рджреМрд░рд╛рди рдЕрдм рддрдХ рджреЗрдЦреА рдЧрдИ рдЫрд╡рд┐рдпреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд░рд┐рдХреЙрд░реНрдб рдХрд░рддрд╛ рд╣реИред |
|
stats |
рд╕рддреНрдпрд╛рдкрди рдХреЗ рджреМрд░рд╛рди рдЖрдВрдХрдбрд╝реЛрдВ рдХреЗ рд▓рд┐рдП рдкреНрд▓реЗрд╕рд╣реЛрд▓реНрдбрд░ред |
|
confusion_matrix |
рдПрдХ рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рд▓рд┐рдП рдкреНрд▓реЗрд╕рд╣реЛрд▓реНрдбрд░ред |
|
nc |
рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред |
|
iouv |
(torch.Tensor): 0.05 рдХреЗ рд░рд┐рдХреНрдд рд╕реНрдерд╛рди рдореЗрдВ 0.50 рд╕реЗ 0.95 рддрдХ IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдбред |
|
jdict |
dict
|
JSON рд╕рддреНрдпрд╛рдкрди рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
speed |
dict
|
рдХреБрдВрдЬреА 'рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕', 'рдЕрдиреБрдорд╛рди', 'рд╣рд╛рдирд┐', 'рдкреЛрд╕реНрдЯрдкреНрд░реЛрд╕реЗрд╕' рдФрд░ рдЙрдирдХреЗ рд╕рдВрдмрдВрдзрд┐рдд рдХреЗ рд╕рд╛рде рд╢рдмреНрджрдХреЛрд╢ рдорд┐рд▓реАрд╕реЗрдХрдВрдб рдореЗрдВ рдмреИрдЪ рдкреНрд░рд╕рдВрд╕реНрдХрд░рдг рд╕рдордпред |
save_dir |
Path
|
рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ред |
plots |
dict
|
рд╡рд┐рдЬрд╝реБрдЕрд▓рд╛рдЗрдЬрд╝реЗрд╢рди рдХреЗ рд▓рд┐рдП рднреВрдЦрдВрдбреЛрдВ рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
callbacks |
dict
|
рд╡рд┐рднрд┐рдиреНрди рдХреЙрд▓рдмреИрдХ рдХрд╛рд░реНрдпреЛрдВ рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/engine/validator.py
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 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 |
|
metric_keys
property
рдореЗрдВ рдЙрдкрдпреЛрдЧ рдХреА рдЧрдИ рдореАрдЯреНрд░рд┐рдХ рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреЛ рд▓реМрдЯрд╛рддрд╛ рд╣реИ YOLO рдкреНрд░рд╢рд┐рдХреНрд╖рдг/рд╕рддреНрдпрд╛рдкрдиред
__call__(trainer=None, model=None)
рдПрдХ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдХреЗ рд╕рддреНрдпрд╛рдкрди рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддрд╛ рд╣реИ рдпрджрд┐ рдЙрддреНрддреАрд░реНрдг рд╣реЛ рдЬрд╛рддрд╛ рд╣реИ рдпрд╛ рдПрдХ рдореЙрдбрд▓ рдХреЛ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд┐рдпрд╛ рдЬрд╛ рд░рд╣рд╛ рд╣реИ рдпрджрд┐ рдЯреНрд░реЗрдирд░ рдкрд╛рд╕ рд╣реЛ рдЬрд╛рддрд╛ рд╣реИ (рдЯреНрд░реЗрдирд░ рдкреНрд░рд╛рдердорд┐рдХрддрд╛ рдорд┐рд▓рддреА рд╣реИ)ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/engine/validator.py
105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217 |
|
__init__(dataloader=None, save_dir=None, pbar=None, args=None, _callbacks=None)
BaseValidator рдЖрд╡реГрддреНрддрд┐ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
dataloader |
DataLoader
|
рд╕рддреНрдпрд╛рдкрди рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рдбреЗрдЯрд╛рд▓реЛрдбрд░ред |
None
|
save_dir |
Path
|
рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ред |
None
|
pbar |
tqdm
|
рдкреНрд░рдЧрддрд┐ рдкреНрд░рджрд░реНрд╢рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдкреНрд░рдЧрддрд┐ рдкрдЯреНрдЯреАред |
None
|
args |
SimpleNamespace
|
рд╕рддреНрдпрд╛рдкрдирдХрд░реНрддрд╛ рдХреЗ рд▓рд┐рдП рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рдиред |
None
|
_callbacks |
dict
|
рд╡рд┐рднрд┐рдиреНрди рдХреЙрд▓рдмреИрдХ рдХрд╛рд░реНрдпреЛрдВ рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
None
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/engine/validator.py
add_callback(event, callback)
build_dataset(img_path)
check_stats(stats)
eval_json(stats)
рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА рдЖрдВрдХрдбрд╝реЛрдВ рдХреЗ JSON рдкреНрд░рд╛рд░реВрдк рдХрд╛ рдореВрд▓реНрдпрд╛рдВрдХрди рдХрд░реЗрдВ рдФрд░ рд╡рд╛рдкрд╕ рдХрд░реЗрдВред
finalize_metrics(*args, **kwargs)
get_dataloader(dataset_path, batch_size)
рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрде рдФрд░ рдмреИрдЪ рдЖрдХрд╛рд░ рд╕реЗ рдбреЗрдЯрд╛ рд▓реЛрдбрд░ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
get_desc()
get_stats()
init_metrics(model)
рдХреЗ рд▓рд┐рдП рдкреНрд░рджрд░реНрд╢рди рдореАрдЯреНрд░рд┐рдХ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВ YOLO рдХреЛ рдЧрдврд╝рдирд╛ред
match_predictions(pred_classes, true_classes, iou, use_scipy=False)
IoU рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдХреА рд╡рд╕реНрддреБрдУрдВ (pred_classes, true_classes) рд╕реЗ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХрд╛ рдорд┐рд▓рд╛рди рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
pred_classes |
Tensor
|
рдЖрдХрд╛рд░ рдХреЗ рдЕрдиреБрдорд╛рдирд┐рдд рд╡рд░реНрдЧ рд╕реВрдЪрдХрд╛рдВрдХ (рдПрди,)ред |
рдЖрд╡рд╢реНрдпрдХ |
true_classes |
Tensor
|
рдЖрдХрд╛рд░ рдХреЗ рд▓рдХреНрд╖реНрдп рд╡рд░реНрдЧ рд╕реВрдЪрдХрд╛рдВрдХ (рдПрдо,)ред |
рдЖрд╡рд╢реНрдпрдХ |
iou |
Tensor
|
рдПрдХ рдПрдирдПрдХреНрд╕рдПрдо tensor рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдФрд░ рд╕рддреНрдп рдХреЗ рдЖрдзрд╛рд░ рдХреЗ рд▓рд┐рдП рдЬреЛрдбрд╝реАрджрд╛рд░ IoU рдореВрд▓реНрдпреЛрдВ рд╕реЗ рдпреБрдХреНрдд |
рдЖрд╡рд╢реНрдпрдХ |
use_scipy |
bool
|
рдорд┐рд▓рд╛рди рдХреЗ рд▓рд┐рдП рд╕реНрдХрд┐рдкреА рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВ (рдЕрдзрд┐рдХ рд╕рдЯреАрдХ)ред |
False
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдЬреА рд╣рд╛рдБ tensor 10 IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП рдЖрдХрд╛рд░ (N, 10)ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/engine/validator.py
on_plot(name, data=None)
рд░рдЬрд┐рд╕реНрдЯрд░, рднреВрдЦрдВрдб (рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП рдХреЙрд▓рдмреИрдХ рдореЗрдВ рдЙрдкрднреЛрдЧ рдХрд┐рдпрд╛ рдЬрд╛рдирд╛)
plot_predictions(batch, preds, ni)
рднреВрдЦрдВрдбреЛрдВ YOLO рдмреИрдЪ рдЫрд╡рд┐рдпреЛрдВ рдкрд░ рдореЙрдбрд▓ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпрд╛рдВред
plot_val_samples(batch, ni)
postprocess(preds)
'рдкреЛрд╕реНрдЯрдкреНрд░реЛрд╕реЗрд╕ ()' рдХреЗ рдЙрджреНрджреЗрд╢реНрдп рдХрд╛ рд╡рд░реНрдгрди рдФрд░ рд╕рд╛рд░рд╛рдВрд╢ рдХрд░рддрд╛ рд╣реИ рд▓реЗрдХрд┐рди рдХреЛрдИ рд╡рд┐рд╡рд░рдг рдирд╣реАрдВ рдмрддрд╛рдпрд╛ рдЧрдпрд╛ рд╣реИред
pred_to_json(preds, batch)
preprocess(batch)
print_results()
рдореЙрдбрд▓ рдХреА рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рдкреНрд░рд┐рдВрдЯ рдХрд░рддрд╛ рд╣реИред
run_callbacks(event)
рдХрд┐рд╕реА рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдИрд╡реЗрдВрдЯ рд╕реЗ рдЬреБрдбрд╝реЗ рд╕рднреА рдХреЙрд▓рдмреИрдХ рдЪрд▓рд╛рддрд╛ рд╣реИред
update_metrics(preds, batch)
рдкреВрд░реНрд╡рд╛рдиреБрдорд╛рдиреЛрдВ рдФрд░ рдмреИрдЪ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдореАрдЯреНрд░рд┐рдХ рдЕрдкрдбреЗрдЯ рдХрд░рддрд╛ рд╣реИред
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (3)