рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/data/augment.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/data/augment.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.data.augment.BaseTransform
рдЫрд╡рд┐ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЗ рд▓рд┐рдП рдмреЗрд╕ рдХреНрд▓рд╛рд╕ред
рдпрд╣ рдПрдХ рд╕рд╛рдорд╛рдиреНрдп рдкрд░рд┐рд╡рд░реНрддрди рд╡рд░реНрдЧ рд╣реИ рдЬрд┐рд╕реЗ рд╡рд┐рд╢рд┐рд╖реНрдЯ рдЫрд╡рд┐ рдкреНрд░рд╕рдВрд╕реНрдХрд░рдг рдЖрд╡рд╢реНрдпрдХрддрд╛рдУрдВ рдХреЗ рд▓рд┐рдП рдмрдврд╝рд╛рдпрд╛ рдЬрд╛ рд╕рдХрддрд╛ рд╣реИред рдХрдХреНрд╖рд╛ рдХреЛ рд╡рд░реНрдЧреАрдХрд░рдг рдФрд░ рдЕрд░реНрде рд╡рд┐рднрд╛рдЬрди рдХрд╛рд░реНрдпреЛрдВ рджреЛрдиреЛрдВ рдХреЗ рд╕рд╛рде рд╕рдВрдЧрдд рд╣реЛрдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
__init__ |
BaseTransform рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред |
apply_image |
рд▓реЗрдмрд▓ рдкрд░ рдЫрд╡рд┐ рдкрд░рд┐рд╡рд░реНрддрди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред |
apply_instances |
рд▓реЗрдмрд▓ рдореЗрдВ рдСрдмреНрдЬреЗрдХреНрдЯ рдЗрдВрд╕реНрдЯреЗрдВрд╕ рдкрд░ рдЯреНрд░рд╛рдВрд╕реНрдлрд╝реЙрд░реНрдореЗрд╢рди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ. |
apply_semantic |
рдПрдХ рдЫрд╡рд┐ рдХреЗ рд▓рд┐рдП рдЕрд░реНрде рд╡рд┐рднрд╛рдЬрди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ. |
__call__ |
рд╕рднреА рд▓реЗрдмрд▓ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЛ рдПрдХ рдЫрд╡рд┐, рдЙрджрд╛рд╣рд░рдг рдФрд░ рд╕рд┐рдореЗрдВрдЯрд┐рдХ рдорд╛рд╕реНрдХ рдкрд░ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рд╕рднреА рд▓реЗрдмрд▓ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЛ рдПрдХ рдЫрд╡рд┐, рдЙрджрд╛рд╣рд░рдг рдФрд░ рд╕рд┐рдореЗрдВрдЯрд┐рдХ рдорд╛рд╕реНрдХ рдкрд░ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
__init__()
apply_image(labels)
apply_instances(labels)
рд▓реЗрдмрд▓ рдореЗрдВ рдСрдмреНрдЬреЗрдХреНрдЯ рдЗрдВрд╕реНрдЯреЗрдВрд╕ рдкрд░ рдЯреНрд░рд╛рдВрд╕реНрдлрд╝реЙрд░реНрдореЗрд╢рди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ.
ultralytics.data.augment.Compose
рдХрдИ рдЫрд╡рд┐ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреА рд░рдЪрдирд╛ рдХреЗ рд▓рд┐рдП рдХрдХреНрд╖рд╛ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(data)
рдЗрдирдкреБрдЯ рдбреЗрдЯрд╛ рдореЗрдВ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреА рдПрдХ рд╢реНрд░реГрдВрдЦрд▓рд╛ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
__getitem__(index)
рдЕрдиреБрдХреНрд░рдордг рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдПрдХ рд╡рд┐рд╢рд┐рд╖реНрдЯ рд░реВрдкрд╛рдВрддрд░рдг рдпрд╛ рд░реВрдкрд╛рдВрддрд░рдгреЛрдВ рдХрд╛ рдПрдХ рд╕реЗрдЯ рдкреБрдирд░реНрдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(transforms)
рдЯреНрд░рд╛рдВрд╕рдлрд╝реЙрд░реНрдо рдХреА рд╕реВрдЪреА рдХреЗ рд╕рд╛рде рдХрдВрдкреЛрдЬрд╝ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
__repr__()
рдСрдмреНрдЬреЗрдХреНрдЯ рдХрд╛ рдПрдХ рд╕реНрдЯреНрд░рд┐рдВрдЧ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рджреЗрддрд╛ рд╣реИред
__setitem__(index, value)
рдЕрдиреБрдХреНрд░рдордг рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдПрдХ рд╡рд┐рд╢рд┐рд╖реНрдЯ рд░реВрдкрд╛рдВрддрд░рдг рдпрд╛ рд░реВрдкрд╛рдВрддрд░рдгреЛрдВ рдХрд╛ рдПрдХ рд╕реЗрдЯ рдкреБрдирд░реНрдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
append(transform)
рдЯреНрд░рд╛рдВрд╕рдлрд╝реЙрд░реНрдо рдХреА рдореМрдЬреВрджрд╛ рд╕реВрдЪреА рдореЗрдВ рдПрдХ рдирдпрд╛ рдЯреНрд░рд╛рдВрд╕рдлрд╝реЙрд░реНрдо рдЬреЛрдбрд╝рддрд╛ рд╣реИред
insert(index, transform)
рдЯреНрд░рд╛рдВрд╕реНрдлрд╝реЙрд░реНрдо рдХреА рдореМрдЬреВрджрд╛ рд╕реВрдЪреА рдореЗрдВ рдПрдХ рдирдпрд╛ рдЯреНрд░рд╛рдВрд╕реНрдлрд╝реЙрд░реНрдо рд╕рдореНрдорд┐рд▓рд┐рдд рдХрд░рддрд╛ рд╣реИ.
tolist()
рд░реВрдкрд╛рдВрддрд░рдгреЛрдВ рдХреА рд╕реВрдЪреА рдХреЛ рдорд╛рдирдХ рдореЗрдВ рдХрдирд╡рд░реНрдЯ рдХрд░рддрд╛ рд╣реИ Python рддрд╛рд▓рд┐рдХрд╛ред
ultralytics.data.augment.BaseMixTransform
рдмреЗрд╕ рдорд┐рдХреНрд╕ (рдорд┐рдХреНрд╕рдЕрдк/рдореЛрдЬрд╝реЗрдХ) рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧред
рдпрд╣ рдХрд╛рд░реНрдпрд╛рдиреНрд╡рдпрди mmyolo рд╕реЗ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рдкреВрд░реНрд╡-рдкреНрд░рд╕рдВрд╕реНрдХрд░рдг рдкрд░рд┐рд╡рд░реНрддрди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ рдФрд░ рд▓реЗрдмрд▓ рдбреЗрдЯрд╛ рдореЗрдВ рдорд┐рдХреНрд╕рдЕрдк /
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(dataset, pre_transform=None, p=0.0)
рдбреЗрдЯрд╛рд╕реЗрдЯ, pre_transform рдФрд░ рд╕рдВрднрд╛рд╡рдирд╛ рдХреЗ рд╕рд╛рде BaseMixTransform рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
get_indexes()
рдореЛрдЬрд╝реЗрдХ рд╡реГрджреНрдзрд┐ рдХреЗ рд▓рд┐рдП рдлреЗрд░рдмрджрд▓ рдЗрдВрдбреЗрдХреНрд╕ рдХреА рдПрдХ рд╕реВрдЪреА рдкреНрд░рд╛рдкреНрдд рдХрд░рддрд╛ рд╣реИред
ultralytics.data.augment.Mosaic
рдХрд╛ рд░реВрдк: BaseMixTransform
рдореЛрдЬрд╝реЗрдХ рд╡реГрджреНрдзрд┐ред
рдпрд╣ рд╡рд░реНрдЧ рдПрдХ рдПрдХрд▓ рдореЛрдЬрд╝реЗрдХ рдЫрд╡рд┐ рдореЗрдВ рдХрдИ (4 рдпрд╛ 9) рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рдЬреЛрдбрд╝рдХрд░ рдореЛрдЬрд╝реЗрдХ рд╡реГрджреНрдзрд┐ рдХрд░рддрд╛ рд╣реИред рд╡реГрджреНрдзрд┐ рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рд╕рдВрднрд╛рд╡рдирд╛ рд╡рд╛рд▓реЗ рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░ рд▓рд╛рдЧреВ рд╣реЛрддреА рд╣реИред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
dataset |
рдбреЗрдЯрд╛рд╕реЗрдЯ рдЬрд┐рд╕ рдкрд░ рдореЛрдЬрд╝реЗрдХ рд╡реГрджреНрдзрд┐ рд▓рд╛рдЧреВ рд╣реЛрддреА рд╣реИред |
|
imgsz |
int
|
рдПрдХ рдЫрд╡рд┐ рдХреА рдореЛрдЬрд╝реЗрдХ рдкрд╛рдЗрдкрд▓рд╛рдЗрди рдХреЗ рдмрд╛рдж рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ (рдКрдВрдЪрд╛рдИ рдФрд░ рдЪреМрдбрд╝рд╛рдИ)ред 640 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
p |
float
|
рдореЛрдЬрд╝реЗрдХ рд╡реГрджреНрдзрд┐ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ред 0-1 рдХреА рд╕реАрдорд╛ рдореЗрдВ рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдПред 1.0 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
n |
int
|
рдЧреНрд░рд┐рдб рдЖрдХрд╛рд░, рдпрд╛ рддреЛ 4 (2x2 рдХреЗ рд▓рд┐рдП) рдпрд╛ 9 (3x3 рдХреЗ рд▓рд┐рдП)ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
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 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 |
|
__init__(dataset, imgsz=640, p=1.0, n=4)
рдбреЗрдЯрд╛рд╕реЗрдЯ, рдЫрд╡рд┐ рдЖрдХрд╛рд░, рд╕рдВрднрд╛рд╡рдирд╛ рдФрд░ рд╕реАрдорд╛ рдХреЗ рд╕рд╛рде рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
get_indexes(buffer=True)
рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдЕрдиреБрдХреНрд░рдордгрд┐рдХрд╛ рдХреА рдПрдХ рд╕реВрдЪреА рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.MixUp
рдХрд╛ рд░реВрдк: BaseMixTransform
рдбреЗрдЯрд╛рд╕реЗрдЯ рдореЗрдВ рдорд┐рдХреНрд╕рдЕрдк рд╡реГрджреНрдзрд┐ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(dataset, pre_transform=None, p=0.0)
рдбреЗрдЯрд╛рд╕реЗрдЯ, pre_transform рдФрд░ рдорд┐рдХреНрд╕рдЕрдк рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ рдХреЗ рд╕рд╛рде рдорд┐рдХреНрд╕рдЕрдк рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
get_indexes()
рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ рдПрдХ рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рд╕реВрдЪрдХрд╛рдВрдХ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
ultralytics.data.augment.RandomPerspective
рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдкрд░рд┐рдкреНрд░реЗрдХреНрд╖реНрдп рдХреЛ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ рдФрд░ рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рд╕рдВрдмрдВрдзрд┐рдд рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, рд╕реЗрдЧрдореЗрдВрдЯ рдФрд░ рдкрд░ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЛ рдареАрдХ рдХрд░рддрд╛ рд╣реИ рдореБрдЦреНрдп рдмрд┐рдВрджреБред рдЗрди рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдореЗрдВ рд░реЛрдЯреЗрд╢рди, рдЕрдиреБрд╡рд╛рдж, рд╕реНрдХреЗрд▓рд┐рдВрдЧ рдФрд░ рдХрддрд░рдиреА рд╢рд╛рдорд┐рд▓ рд╣реИрдВред рдХрдХреНрд╖рд╛ рднреА рдкреНрд░рджрд╛рди рдХрд░рддреА рд╣реИ рдПрдХ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рд╕рдВрднрд╛рд╡рдирд╛ рдХреЗ рд╕рд╛рде рдЗрди рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЛ рд╕рд╢рд░реНрдд рд░реВрдк рд╕реЗ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХрд╛ рд╡рд┐рдХрд▓реНрдкред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
degrees |
float
|
рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдШреБрдорд╛рд╡ рдХреЗ рд▓рд┐рдП рдбрд┐рдЧреНрд░реА рд░реЗрдВрдЬред |
translate |
float
|
рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдЕрдиреБрд╡рд╛рдж рдХреЗ рд▓рд┐рдП рдХреБрд▓ рдЪреМрдбрд╝рд╛рдИ рдФрд░ рдКрдВрдЪрд╛рдИ рдХрд╛ рдЕрдВрд╢ред |
scale |
float
|
рд╕реНрдХреЗрд▓рд┐рдВрдЧ рдХрд╛рд░рдХ рдЕрдВрддрд░рд╛рд▓, рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП, 0.1 рдХрд╛ рдПрдХ рд╕реНрдХреЗрд▓ рдХрд╛рд░рдХ 90% -110% рдХреЗ рдмреАрдЪ рдЖрдХрд╛рд░ рдмрджрд▓рдиреЗ рдХреА рдЕрдиреБрдорддрд┐ рджреЗрддрд╛ рд╣реИред |
shear |
float
|
рдХрддрд░рдиреА рддреАрд╡реНрд░рддрд╛ (рдбрд┐рдЧреНрд░реА рдореЗрдВ рдХреЛрдг)ред |
perspective |
float
|
рдкрд░рд┐рдкреНрд░реЗрдХреНрд╖реНрдп рд╡рд┐рд░реВрдкрдг рдХрд╛рд░рдХред |
border |
tuple
|
рдореЛрдЬрд╝реЗрдХ рд╕реАрдорд╛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдХрд░рдиреЗ рд╡рд╛рд▓рд╛ рдЯрдкрд▓ред |
pre_transform |
callable
|
рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдкрд░рд┐рд╡рд░реНрддрди рд╢реБрд░реВ рдХрд░рдиреЗ рд╕реЗ рдкрд╣рд▓реЗ рдЫрд╡рд┐ рдкрд░ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдлрд╝рдВрдХреНрд╢рди / |
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
affine_transform |
рдЫрд╡рд┐ рдХреЗ рд▓рд┐рдП affine рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреА рдПрдХ рд╢реНрд░реГрдВрдЦрд▓рд╛ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред |
apply_bboxes |
рдкрд░рд┐рдХрд▓рд┐рдд affine рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░ bounding рдмрдХреНрд╕реЗ рдХреЛ рдмрджрд▓ рджреЗрддрд╛ рд╣реИ. |
apply_segments |
рдЦрдВрдбреЛрдВ рдХреЛ рдмрджрд▓ рджреЗрддрд╛ рд╣реИ рдФрд░ рдирдП рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдЙрддреНрдкрдиреНрди рдХрд░рддрд╛ рд╣реИред |
apply_keypoints |
рдореБрдЦреНрдп рдмрд┐рдВрджреБрдУрдВ рдХреЛ рдмрджрд▓ рджреЗрддрд╛ рд╣реИред |
__call__ |
рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рдЙрдирдХреЗ рд╕рдВрдмрдВрдзрд┐рдд рдПрдиреЛрдЯреЗрд╢рди рджреЛрдиреЛрдВ рдореЗрдВ рдкрд░рд┐рд╡рд░реНрддрди рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреА рдореБрдЦреНрдп рд╡рд┐рдзрд┐ред |
box_candidates |
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЛ рдлрд╝рд┐рд▓реНрдЯрд░ рдХрд░рддрд╛ рд╣реИ рдЬреЛ рдкрд░рд┐рд╡рд░реНрддрди рдХреЗ рдмрд╛рдж рдХреБрдЫ рдорд╛рдирджрдВрдбреЛрдВ рдХреЛ рдкреВрд░рд╛ рдирд╣реАрдВ рдХрд░рддреЗ рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 |
|
__call__(labels)
рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рд▓рдХреНрд╖реНрдпреЛрдВ рдХреЛ рдареАрдХ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
labels |
dict
|
рдХрд╛ рдПрдХ рд╣реБрдХреНрдо |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, border=(0, 0), pre_transform=None)
рдкрд░рд┐рд╡рд░реНрддрди рдорд╛рдкрджрдВрдбреЛрдВ рдХреЗ рд╕рд╛рде RandomPerspective рдСрдмреНрдЬреЗрдХреНрдЯ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
affine_transform(img, border)
рдЫрд╡рд┐ рдХреЗрдВрджреНрд░ рдХреЗ рдЖрд╕рдкрд╛рд╕ рдХреЗрдВрджреНрд░рд┐рдд affine рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХрд╛ рдПрдХ рдХреНрд░рдо рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
img |
ndarray
|
рдЗрдирдкреБрдЯ рдЫрд╡рд┐ред |
рдЖрд╡рд╢реНрдпрдХ |
border |
tuple
|
рдмреЙрд░реНрдбрд░ рдЖрдпрд╛рдо. |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
img |
ndarray
|
рд░реВрдкрд╛рдВрддрд░рд┐рдд рдЫрд╡рд┐ред |
M |
ndarray
|
рдкрд░рд┐рд╡рд░реНрддрди рдореИрдЯреНрд░рд┐рдХреНрд╕ред |
s |
float
|
рд╕реНрдХреЗрд▓ рдХрд╛рд░рдХред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
apply_bboxes(bboxes, M)
рдХреЗрд╡рд▓ bboxes рдкрд░ affine рд▓рд╛рдЧреВ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
bboxes |
ndarray
|
bboxes рдХреА рд╕реВрдЪреА, xyxy рдкреНрд░рд╛рд░реВрдк, рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде (num_bboxes, 4). |
рдЖрд╡рд╢реНрдпрдХ |
M |
ndarray
|
рдПрдлрд┐рди рдореИрдЯреНрд░рд┐рдХреНрд╕ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
new_bboxes |
ndarray
|
рдПрдлрд┐рди рдХреЗ рдмрд╛рдж рдмреАрдмреЙрдХреНрд╕, [num_bboxes, 4]ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
apply_keypoints(keypoints, M)
рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдкрд░ рдПрдлрд╛рдЗрди рд▓рдЧрд╛рдПрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
keypoints |
ndarray
|
рдХреАрдкреЙрдЗрдВрдЯреНрд╕, [рдПрди, 17, 3]ред |
рдЖрд╡рд╢реНрдпрдХ |
M |
ndarray
|
рдПрдлрд┐рди рдореИрдЯреНрд░рд┐рдХреНрд╕ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
new_keypoints |
ndarray
|
рдПрдлрд┐рди рдХреЗ рдмрд╛рдж рдХреАрдкреЙрдЗрдВрдЯ, [рдПрди, 17, 3]ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
apply_segments(segments, M)
рд╕реЗрдЧрдореЗрдВрдЯ рдореЗрдВ affine рд▓рд╛рдЧреВ рдХрд░реЗрдВ рдФрд░ рд╕реЗрдЧрдореЗрдВрдЯ рд╕реЗ рдирдП bboxes рдЙрддреНрдкрдиреНрди рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
segments |
ndarray
|
рдЦрдВрдбреЛрдВ рдХреА рд╕реВрдЪреА, [num_samples, 500, 2]ред |
рдЖрд╡рд╢реНрдпрдХ |
M |
ndarray
|
рдПрдлрд┐рди рдореИрдЯреНрд░рд┐рдХреНрд╕ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
new_segments |
ndarray
|
рдПрдлрд┐рди рдХреЗ рдмрд╛рдж рдЦрдВрдбреЛрдВ рдХреА рд╕реВрдЪреА, [num_samples, 500, 2]ред |
new_bboxes |
ndarray
|
рдПрдлрд┐рди рдХреЗ рдмрд╛рдж рдмрдХреНрд╕реЗ, [рдПрди, 4]ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
box_candidates(box1, box2, wh_thr=2, ar_thr=100, area_thr=0.1, eps=1e-16)
рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХреЗ рдПрдХ рд╕реЗрдЯ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдмреЙрдХреНрд╕ рдЙрдореНрдореАрджрд╡рд╛рд░реЛрдВ рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред рдпрд╣ рд╡рд┐рдзрд┐ рдмрдХреНрд╕реЗ рдХреА рд╡рд┐рд╢реЗрд╖рддрд╛рдУрдВ рдХреА рддреБрд▓рдирд╛ рдХрд░рддреА рд╣реИ рд╡реГрджреНрдзрд┐ рд╕реЗ рдкрд╣рд▓реЗ рдФрд░ рдмрд╛рдж рдореЗрдВ рдпрд╣ рддрдп рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдХрд┐ рдХреНрдпрд╛ рдПрдХ рдмреЙрдХреНрд╕ рдЖрдЧреЗ рдХреА рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдЙрдореНрдореАрджрд╡рд╛рд░ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
box1 |
ndarray
|
рд╡реГрджреНрдзрд┐ рд╕реЗ рдкрд╣рд▓реЗ 4,n рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, [x1, y1, x2, y2] рдХреЗ рд░реВрдк рдореЗрдВ рджрд░реНрд╢рд╛рдпрд╛ рдЧрдпрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
box2 |
ndarray
|
рд╡реГрджреНрдзрд┐ рдХреЗ рдмрд╛рдж 4, n рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, [x1, y1, x2, y2] рдХреЗ рд░реВрдк рдореЗрдВ рджрд░реНрд╢рд╛рдпрд╛ рдЧрдпрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
wh_thr |
float
|
рдкрд┐рдХреНрд╕реЗрд▓ рдореЗрдВ рдЪреМрдбрд╝рд╛рдИ рдФрд░ рдКрдВрдЪрд╛рдИ рдереНрд░реЗрд╢реЛрд▓реНрдбред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 2 рд╣реИред |
2
|
ar_thr |
float
|
рдкрд╣рд▓реВ рдЕрдиреБрдкрд╛рдд рд╕реАрдорд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 100 рд╣реИред |
100
|
area_thr |
float
|
рдХреНрд╖реЗрддреНрд░ рдЕрдиреБрдкрд╛рдд рджрд╣рд▓реАрдЬред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.1 рд╣реИред |
0.1
|
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рдХреЛ рд░реЛрдХрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдПрдкреНрд╕рд┐рд▓реЙрди рдорд╛рдиред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1e-16 рд╣реИред |
1e-16
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рдПрдХ рдмреВрд▓рд┐рдпрди рд╕рд░рдгреА рдпрд╣ рджрд░реНрд╢рд╛рддреА рд╣реИ рдХрд┐ рджрд┐рдП рдЧрдП рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдХреМрди рд╕реЗ рдмреЙрдХреНрд╕ рдЙрдореНрдореАрджрд╡рд╛рд░ рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.RandomHSV
рдпрд╣ рд╡рд░реНрдЧ рдПрдХ рдХреЗ рд╣реНрдпреВ, рд╕рдВрддреГрдкреНрддрд┐ рдФрд░ рдореВрд▓реНрдп (рдПрдЪрдПрд╕рд╡реА) рдЪреИрдирд▓реЛрдВ рдХреЗ рд▓рд┐рдП рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рд╕рдорд╛рдпреЛрдЬрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЬрд┐рдореНрдореЗрджрд╛рд░ рд╣реИ рдкреНрд░рддрд┐рдмрд┐рдВрдмред
рд╕рдорд╛рдпреЛрдЬрди рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рд╣реИрдВ рд▓реЗрдХрд┐рди hgain, sgain рдФрд░ vgain рджреНрд╡рд╛рд░рд╛ рдирд┐рд░реНрдзрд╛рд░рд┐рдд рд╕реАрдорд╛рдУрдВ рдХреЗ рднреАрддрд░ рд╣реИрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рдкреВрд░реНрд╡рдирд┐рд░реНрдзрд╛рд░рд┐рдд рд╕реАрдорд╛рдУрдВ рдХреЗ рднреАрддрд░ рдПрдХ рдЫрд╡рд┐ рдХреЗ рд▓рд┐рдП рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдПрдЪрдПрд╕рд╡реА рд╡реГрджреНрдзрд┐ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
рд╕рдВрд╢реЛрдзрд┐рдд рдЫрд╡рд┐ рдЗрдирдкреБрдЯ 'рд▓реЗрдмрд▓' рдбрд┐рдХреНрдЯ рдореЗрдВ рдореВрд▓ рдЫрд╡рд┐ рдХреЛ рдмрджрд▓ рджреЗрддреА рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(hgain=0.5, sgain=0.5, vgain=0.5)
рдкреНрд░рддреНрдпреЗрдХ HSV рдЪреИрдирд▓ рдХреЗ рд▓рд┐рдП рд▓рд╛рдн рдХреЗ рд╕рд╛рде RandomHSV рд╡рд░реНрдЧ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
hgain |
float
|
рд░рдВрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо рднрд┐рдиреНрдирддрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
sgain |
float
|
рд╕рдВрддреГрдкреНрддрд┐ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо рднрд┐рдиреНрдирддрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
vgain |
float
|
рдорд╛рди рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо рднрд┐рдиреНрдирддрд╛. рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.RandomFlip
рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рд╕рдВрднрд╛рд╡рдирд╛ рдХреЗ рд╕рд╛рде рдПрдХ рдЫрд╡рд┐ рдХреЗ рд▓рд┐рдП рдПрдХ рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдХреНрд╖реИрддрд┐рдЬ рдпрд╛ рдКрд░реНрдзреНрд╡рд╛рдзрд░ рдлреНрд▓рд┐рдк рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
рддрджрдиреБрд╕рд╛рд░ рдХрд┐рд╕реА рднреА рдЙрджрд╛рд╣рд░рдг (рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, рдХреАрдкреЙрдЗрдВрдЯ, рдЖрджрд┐) рдХреЛ рднреА рдЕрдкрдбреЗрдЯ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рдПрдХ рдЫрд╡рд┐ рдХреЗ рд▓рд┐рдП рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдлреНрд▓рд┐рдк рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ рдФрд░ рддрджрдиреБрд╕рд╛рд░ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдпрд╛ рдХреАрдкреЙрдЗрдВрдЯ рдЬреИрд╕реЗ рдХрд┐рд╕реА рднреА рдЙрджрд╛рд╣рд░рдг рдХреЛ рдЕрдкрдбреЗрдЯ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
labels |
dict
|
рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рдЬрд┐рд╕рдореЗрдВ 'img' рдФрд░ 'рдЙрджрд╛рд╣рд░рдг' рдХреБрдВрдЬрд┐рдпрд╛рдБ рд╣реЛрддреА рд╣реИрдВред 'img' рдлрд╝реНрд▓рд┐рдк рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐ рд╣реИред 'рдЙрджрд╛рд╣рд░рдг' рдПрдХ рдСрдмреНрдЬреЗрдХреНрдЯ рд╣реИ рдЬрд┐рд╕рдореЗрдВ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡реИрдХрд▓реНрдкрд┐рдХ рд░реВрдк рд╕реЗ рдХреАрдкреЙрдЗрдВрдЯ рд╣реЛрддреЗ рд╣реИрдВред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
dict
|
рдлрд╝реНрд▓рд┐рдк рдХреА рдЧрдИ рдЫрд╡рд┐ рдХреЗ рд╕рд╛рде рдПрдХ рд╣реА рд╣реБрдХреНрдо рдФрд░ 'img' рдФрд░ 'рдЙрджрд╛рд╣рд░рдг' рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреЗ рддрд╣рдд рдЕрджреНрдпрддрди рдЙрджрд╛рд╣рд░рдгред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(p=0.5, direction='horizontal', flip_idx=None)
рд╕рдВрднрд╛рд╡рдирд╛ рдФрд░ рджрд┐рд╢рд╛ рдХреЗ рд╕рд╛рде RandomFlip рд╡рд░реНрдЧ рдХреЛ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
p |
float
|
рдлреНрд▓рд┐рдк рд▓рдЧрд╛рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ред 0 рдФрд░ 1 рдХреЗ рдмреАрдЪ рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдП. рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
direction |
str
|
рдлреНрд▓рд┐рдк рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреА рджрд┐рд╢рд╛ред "рдХреНрд╖реИрддрд┐рдЬ" рдпрд╛ "рдКрд░реНрдзреНрд╡рд╛рдзрд░" рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдП. рдбрд┐рдлрд╝реЙрд▓реНрдЯ "рдХреНрд╖реИрддрд┐рдЬ" рд╣реИред |
'horizontal'
|
flip_idx |
array - like
|
рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдлрд╝реНрд▓рд┐рдк рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЗрдВрдбреЗрдХреНрд╕ рдореИрдкрд┐рдВрдЧ, рдпрджрд┐ рдХреЛрдИ рд╣реЛред |
None
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.LetterBox
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ рдмрджрд▓реЗрдВ рдФрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ, рдЙрджрд╛рд╣рд░рдг рд╡рд┐рднрд╛рдЬрди, рдореБрджреНрд░рд╛ рдХреЗ рд▓рд┐рдП рдкреИрдбрд┐рдВрдЧред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels=None, image=None)
рдЕрджреНрдпрддрди рд▓реЗрдмрд▓ рдФрд░ рдЫрд╡рд┐ рдХреЛ рдЕрддрд┐рд░рд┐рдХреНрдд рд╕реАрдорд╛ рдХреЗ рд╕рд╛рде рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(new_shape=(640, 640), auto=False, scaleFill=False, scaleup=True, center=True, stride=32)
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдорд╛рдкрджрдВрдбреЛрдВ рдХреЗ рд╕рд╛рде рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рдСрдмреНрдЬреЗрдХреНрдЯ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.CopyPaste
рдкреЗрдкрд░ https://arxiv.org/abs/2012.07177 рдореЗрдВ рд╡рд░реНрдгрд┐рдд рдХреЙрдкреА-рдкреЗрд╕реНрдЯ рд╡реГрджреНрдзрд┐ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред рдпрд╣ рд╡рд░реНрдЧ рд╣реИ рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рдЙрдирдХреЗ рд╕рдВрдмрдВрдзрд┐рдд рдЙрджрд╛рд╣рд░рдгреЛрдВ рдкрд░ рдХреЙрдкреА-рдкреЗрд╕реНрдЯ рд╡реГрджреНрдзрд┐ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЬрд┐рдореНрдореЗрджрд╛рд░ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рджрд┐рдП рдЧрдП рдЪрд┐рддреНрд░ рдФрд░ рдЙрджрд╛рд╣рд░рдгреЛрдВ рдкрд░ рдХреЙрдкреА-рдкреЗрд╕реНрдЯ рд╡реГрджреНрдзрд┐ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
labels |
dict
|
рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рдЬрд┐рд╕рдореЗрдВ рд╢рд╛рдорд┐рд▓ рд╣реИрдВ: - 'img': рд╡реГрджреНрдзрд┐ рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ред - 'cls': рдЗрдВрд╕реНрдЯреЗрдВрд╕ рд╕реЗ рдЬреБрдбрд╝реЗ рдХреНрд▓рд╛рд╕ рд▓реЗрдмрд▓. - 'рдЙрджрд╛рд╣рд░рдг': рдСрдмреНрдЬреЗрдХреНрдЯ рдЬрд┐рд╕рдореЗрдВ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рд╣реЛрддреЗ рд╣реИрдВ, рдФрд░ рд╡реИрдХрд▓реНрдкрд┐рдХ рд░реВрдк рд╕реЗ, рдХреАрдкреЙрдЗрдВрдЯ рдФрд░ рд╕реЗрдЧрдореЗрдВрдЯред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
dict
|
рд╕рдВрд╡рд░реНрдзрд┐рдд рдЫрд╡рд┐ рдФрд░ 'img', 'cls', рдФрд░ 'instances' рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреЗ рддрд╣рдд рдЕрдкрдбреЗрдЯ рдХрд┐рдП рдЧрдП рдЙрджрд╛рд╣рд░рдгреЛрдВ рдХреЗ рд╕рд╛рде рдбрд┐рдХреНрдЯ рдХрд░реЗрдВред |
рдиреЛрдЯреНрд╕
- рдЙрджрд╛рд╣рд░рдгреЛрдВ рд╕реЗ рдЗрд╕ рд╡реГрджреНрдзрд┐ рдХреЗ рд▓рд┐рдП рдХрд╛рдо рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдирдХреА рд╡рд┐рд╢реЗрд╖рддрд╛рдУрдВ рдореЗрдВ рд╕реЗ рдПрдХ рдХреЗ рд░реВрдк рдореЗрдВ 'рдЦрдВрдб' рд╣реЛрдиреЗ рдХреА рдЙрдореНрдореАрдж рд╣реИред
- рдпрд╣ рд╡рд┐рдзрд┐ рдЗрдирдкреБрдЯ рд╢рдмреНрджрдХреЛрд╢ 'рд▓реЗрдмрд▓' рдХреЛ рдЬрдЧрд╣ рдореЗрдВ рд╕рдВрд╢реЛрдзрд┐рдд рдХрд░рддреА рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(p=0.5)
рджреА рдЧрдИ рдкреНрд░рд╛рдпрд┐рдХрддрд╛ рдХреЗ рд╕рд╛рде CopyPaste рд╡рд░реНрдЧ рдХреЛ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
p |
float
|
рдХреЙрдкреА-рдкреЗрд╕реНрдЯ рд╡реГрджреНрдзрд┐ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ред 0 рдФрд░ 1 рдХреЗ рдмреАрдЪ рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдП. рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.Albumentations
рдПрд▓реНрдмрдореЗрдВрдЯреЗрд╢рди рдкрд░рд┐рд╡рд░реНрддрдиред
рд╡реИрдХрд▓реНрдкрд┐рдХ, рдЕрдХреНрд╖рдо рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдкреИрдХреЗрдЬ рдХреА рд╕реНрдерд╛рдкрдирд╛ рд░рджреНрдж рдХрд░реЗрдВред рдмреНрд▓рд░, рдореАрдбрд┐рдпрди рдмреНрд▓рд░ рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИ, рдЧреНрд░реЗрд╕реНрдХреЗрд▓ рдореЗрдВ рдХрдирд╡рд░реНрдЯ рдХрд░рддрд╛ рд╣реИ, рдХрдВрдЯреНрд░рд╛рд╕реНрдЯ рд▓рд┐рдорд┐рдЯреЗрдб рдПрдбреЗрдкреНрдЯрд┐рд╡ рд╣рд┐рд╕реНрдЯреЛрдЧреНрд░рд╛рдо рд╕рдореАрдХрд░рдг, рдЪрдордХ рдФрд░ рдХрдВрдЯреНрд░рд╛рд╕реНрдЯ рдХрд╛ рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдкрд░рд┐рд╡рд░реНрддрди, рд░реИрдВрдбрдордЧрд╛рдорд╛ рдФрд░ рдЫрд╡рд┐ рдЧреБрдгрд╡рддреНрддрд╛ рдХреЛ рдХрдо рдХрд░рдирд╛ рд╕рдВрдкреАрдбрдиред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(labels)
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдЙрддреНрдкрдиреНрди рдХрд░рддрд╛ рд╣реИ рдФрд░ рдбрд┐рдЯреЗрдХреНрд╢рди рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЗ рд╕рд╛рде рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(p=1.0)
рдХреЗ рд▓рд┐рдП рдЯреНрд░рд╛рдВрд╕рдлрд╝реЙрд░реНрдо рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВ YOLO bbox рд╕реНрд╡рд░реВрдкрд┐рдд params.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.Format
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди, рдЗрдВрд╕реНрдЯреЗрдВрд╕ рд╕реЗрдЧрдореЗрдВрдЯреЗрд╢рди рдФрд░ рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рдПрдиреЛрдЯреЗрд╢рди рдХреЛ рд╕реНрд╡рд░реВрдкрд┐рдд рдХрд░рддрд╛ рд╣реИред рдХрдХреНрд╖рд╛
рджреНрд╡рд╛рд░рд╛ рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐ рдФрд░ рдЙрджрд╛рд╣рд░рдг рдПрдиреЛрдЯреЗрд╢рди рдХреЛ рдорд╛рдирдХреАрдХреГрдд рдХрд░рддрд╛ рд╣реИ collate_fn
рдореЗрдВ PyTorch рдбреЗрдЯрд╛рд▓реЛрдбрд░ред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
bbox_format |
str
|
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЗ рд▓рд┐рдП рдкреНрд░рд╛рд░реВрдкред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 'xywh' рд╣реИред |
normalize |
bool
|
рдХреНрдпрд╛ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЛ рд╕рд╛рдорд╛рдиреНрдп рдХрд░рдирд╛ рд╣реИред рдбрд┐рдлрд╝реЙрд▓реНрдЯ True рд╣реИ. |
return_mask |
bool
|
рд╡рд┐рднрд╛рдЬрди рдХреЗ рд▓рд┐рдП рдЙрджрд╛рд╣рд░рдг рдорд╛рд╕реНрдХ рд▓реМрдЯрд╛рдПрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
return_keypoint |
bool
|
рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдореБрдЦреНрдп рдмрд┐рдВрджреБ рд▓реМрдЯрд╛рдПрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
mask_ratio |
int
|
рдорд╛рд╕реНрдХ рдХреЗ рд▓рд┐рдП рдбрд╛рдЙрдирд╕реИрдореНрдкрд▓ рдЕрдиреБрдкрд╛рддред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 4 рд╣реИред |
mask_overlap |
bool
|
рдЪрд╛рд╣реЗ рдорд╛рд╕реНрдХ рдХреЛ рдУрд╡рд░рд▓реИрдк рдХрд░рдирд╛ рд╣реЛред рдбрд┐рдлрд╝реЙрд▓реНрдЯ True рд╣реИ. |
batch_idx |
bool
|
рдмреИрдЪ рдЗрдВрдбреЗрдХреНрд╕ рд░рдЦреЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ True рд╣реИ. |
bgr |
float
|
рдмреАрдЬреАрдЖрд░ рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рд╡рд╛рдкрд╕ рдХрд░рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.0 рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 |
|
__call__(labels)
рд╕реНрд╡рд░реВрдкрд┐рдд рдЫрд╡рд┐, рдХрдХреНрд╖рд╛рдПрдВ, рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ 'collate_fn' рджреНрд╡рд╛рд░рд╛ рдЙрдкрдпреЛрдЧ рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(bbox_format='xywh', normalize=True, return_mask=False, return_keypoint=False, return_obb=False, mask_ratio=4, mask_overlap=True, batch_idx=True, bgr=0.0)
рджрд┐рдП рдЧрдП рдорд╛рдкрджрдВрдбреЛрдВ рдХреЗ рд╕рд╛рде рдкреНрд░рд╛рд░реВрдк рд╡рд░реНрдЧ рдХреЛ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.RandomLoadText
рдмреЗрддрд░рддреАрдм рдврдВрдЧ рд╕реЗ рд╕рдХрд╛рд░рд╛рддреНрдордХ рдЧреНрд░рдВрдереЛрдВ рдФрд░ рдирдХрд╛рд░рд╛рддреНрдордХ рдЧреНрд░рдВрдереЛрдВ рдХрд╛ рдирдореВрдирд╛ рд▓реЗрдВ рдФрд░ рдирдореВрдиреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рдХреЗ рдЕрдиреБрд╕рд╛рд░ рд╡рд░реНрдЧ рд╕реВрдЪрдХрд╛рдВрдХреЛрдВ рдХреЛ рдЕрдкрдбреЗрдЯ рдХрд░реЗрдВред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
prompt_format |
str
|
рд╢реАрдШреНрд░ рдХреЗ рд▓рд┐рдП рдкреНрд░рд╛рд░реВрдкред рдбрд┐рдлрд╝реЙрд▓реНрдЯ '{}' рд╣реИред |
neg_samples |
tuple[int]
|
рдирдХрд╛рд░рд╛рддреНрдордХ рдЧреНрд░рдВрдереЛрдВ рдХреЛ рдмреЗрддрд░рддреАрдм рдврдВрдЧ рд╕реЗ рдирдореВрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд░реЗрдВрдЬрд░, рдбрд┐рдлрд╝реЙрд▓реНрдЯ (80, 80) рд╣реИред |
max_samples |
int
|
рдПрдХ рдЫрд╡рд┐ рдореЗрдВ рд╡рд┐рднрд┐рдиреНрди рдкрд╛рда рдирдореВрдиреЛрдВ рдХреА рдЕрдзрд┐рдХрддрдо рд╕рдВрдЦреНрдпрд╛, рдбрд┐рдлрд╝реЙрд▓реНрдЯ 80 рд╣реИред |
padding |
bool
|
рдЪрд╛рд╣реЗ max_samples рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЧреНрд░рдВрдереЛрдВ рдкреИрдб рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдПред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
padding_value |
str
|
рдкреИрдбрд┐рдВрдЧ рдЯреЗрдХреНрд╕реНрдЯред рдбрд┐рдлрд╝реЙрд▓реНрдЯ "" рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 |
|
__call__(labels)
рдЕрджреНрдпрддрди рдХрдХреНрд╖рд╛рдПрдВ рдФрд░ рдЧреНрд░рдВрде рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(prompt_format='{}', neg_samples=(80, 80), max_samples=80, padding=False, padding_value='')
рджрд┐рдП рдЧрдП рдкреИрд░рд╛рдореАрдЯрд░ рдХреЗ рд╕рд╛рде RandomLoadText рдХреНрд▓рд╛рд╕ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.ClassifyLetterBox
YOLOv8 рдЫрд╡рд┐ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рдХреНрд▓рд╛рд╕, рдПрдХ рдкрд░рд┐рд╡рд░реНрддрди рдкрд╛рдЗрдкрд▓рд╛рдЗрди рдХрд╛ рд╣рд┐рд╕реНрд╕рд╛ рдмрдирдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдпрд╛ рдЧрдпрд╛, рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП, T.Compose([рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ (рдЖрдХрд╛рд░), ToTensor ()])ред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
h |
int
|
рдЫрд╡рд┐ рдХреА рд▓рдХреНрд╖реНрдп рдКрдБрдЪрд╛рдИ. |
w |
int
|
рдЫрд╡рд┐ рдХреА рд▓рдХреНрд╖реНрдп рдЪреМрдбрд╝рд╛рдИ. |
auto |
bool
|
рдпрджрд┐ рд╕рдЪ рд╣реИ, рддреЛ рд╕реНрдЯреНрд░рд╛рдЗрдб рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рд╢реЙрд░реНрдЯ рд╕рд╛рдЗрдб рдХреЗ рд▓рд┐рдП рд╕реНрд╡рдЪрд╛рд▓рд┐рдд рд░реВрдк рд╕реЗ рд╣рд▓ рд╣реЛ рдЬрд╛рддрд╛ рд╣реИред |
stride |
int
|
рд╕реНрдЯреНрд░рд╛рдЗрдб рдорд╛рди, рдЬрд┐рд╕рдХрд╛ рдЙрдкрдпреЛрдЧ рддрдм рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ рдЬрдм 'рдСрдЯреЛ' рд╕рддреНрдп рд╣реЛрддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(im)
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ рдмрджрд▓рддрд╛ рд╣реИ рдФрд░ рдЗрд╕реЗ рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рд╡рд┐рдзрд┐ рдХреЗ рд╕рд╛рде рдкреИрдб рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
ndarray
|
рдЖрдХрд╛рд░ HWC рдХреА рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА рдХреЗ рд░реВрдк рдореЗрдВ рдЗрдирдкреБрдЯ рдЫрд╡рд┐ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рдФрд░ рдЖрдХрд╛рд░ рдмрджрд▓рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐ рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА рдХреЗ рд░реВрдк рдореЗрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(size=(640, 640), auto=False, stride=32)
ClassifyLetterBox рд╡рд░реНрдЧ рдХреЛ рд▓рдХреНрд╖реНрдп рдЖрдХрд╛рд░, рдСрдЯреЛ-рдлрд╝реНрд▓реИрдЧ рдФрд░ рд╕реНрдЯреНрд░рд╛рдЗрдб рдХреЗ рд╕рд╛рде рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
size |
Union[int, Tuple[int, int]]
|
рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рдХреЗ рд▓рд┐рдП рд▓рдХреНрд╖реНрдп рдЖрдпрд╛рдо (рдКрдВрдЪрд╛рдИ, рдЪреМрдбрд╝рд╛рдИ)ред |
(640, 640)
|
auto |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рд╕реНрд╡рдЪрд╛рд▓рд┐рдд рд░реВрдк рд╕реЗ рд╕реНрдЯреНрд░рд╛рдЗрдб рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдЫреЛрдЯреА рднреБрдЬрд╛ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред |
False
|
stride |
int
|
рд╕реНрдЯреНрд░рд╛рдЗрдб рдорд╛рди, рдЬрд┐рд╕рдХрд╛ рдЙрдкрдпреЛрдЧ рддрдм рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ рдЬрдм 'рдСрдЯреЛ' рд╕рддреНрдп рд╣реЛрддрд╛ рд╣реИред |
32
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.CenterCrop
YOLOv8 рдЫрд╡рд┐ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рд╕реЗрдВрдЯрд░рдХреНрд░реЙрдк рдХреНрд▓рд╛рд╕, рдПрдХ рдкрд░рд┐рд╡рд░реНрддрди рдкрд╛рдЗрдкрд▓рд╛рдЗрди рдХрд╛ рд╣рд┐рд╕реНрд╕рд╛ рдмрдирдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдпрд╛ рдЧрдпрд╛, рдЙрджрд╛ред T.Compose([CenterCrop(size), ToTensor ()])ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(im)
рд▓реЗрдЯрд░рдмреЙрдХреНрд╕ рд╡рд┐рдзрд┐ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЫрд╡рд┐ рдХреЗ рдХреЗрдВрджреНрд░ рдХрд╛ рдЖрдХрд╛рд░ рдмрджрд▓рддрд╛ рд╣реИ рдФрд░ рдХреНрд░реЙрдк рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
ndarray
|
рдЖрдХрд╛рд░ HWC рдХреА рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА рдХреЗ рд░реВрдк рдореЗрдВ рдЗрдирдкреБрдЯ рдЫрд╡рд┐ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рдХреЗрдВрджреНрд░-рдХреНрд░реЙрдк рдФрд░ рдЖрдХрд╛рд░ рдмрджрд▓рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐ рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА рдХреЗ рд░реВрдк рдореЗрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(size=640)
рдЫрд╡рд┐ рдХреЛ numpy рд╕рд░рдгреА рд╕реЗ PyTorch tensor.
ultralytics.data.augment.ToTensor
YOLOv8 рдЗрдореЗрдЬ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП ToTensor рдХреНрд▓рд╛рд╕, рдпрд╛рдиреА, T.Compose([LetterBox(size), ToTensor ()])ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__call__(im)
рдПрдХ рдЫрд╡рд┐ рдХреЛ рдПрдХ numpy рд╕рд░рдгреА рд╕реЗ рдПрдХ рдореЗрдВ рдмрджрд▓ рджреЗрддрд╛ рд╣реИ PyTorch tensor, рд╡реИрдХрд▓реНрдкрд┐рдХ рдЕрд░реНрдз-рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рдФрд░ рд╕рд╛рдорд╛рдиреНрдпреАрдХрд░рдг рд▓рд╛рдЧреВ рдХрд░рдирд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
ndarray
|
рдмреАрдЬреАрдЖрд░ рдХреНрд░рдо рдореЗрдВ рдЖрдХрд╛рд░ (рдПрдЪ, рдбрдмреНрд▓реНрдпреВ, рд╕реА) рдХреЗ рд╕рд╛рде рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА рдХреЗ рд░реВрдк рдореЗрдВ рдЗрдирдкреБрдЯ рдЫрд╡рд┐ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ рдХреЗ рд░реВрдк рдореЗрдВ рд░реВрдкрд╛рдВрддрд░рд┐рдд рдЫрд╡рд┐ PyTorch tensor рдлреНрд▓реЛрдЯ 32 рдпрд╛ рдлреНрд▓реЛрдЯ 16 рдореЗрдВ, [0, 1] рдХреЗ рд▓рд┐рдП рд╕рд╛рдорд╛рдиреНрдпреАрдХреГрддред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
__init__(half=False)
рдкреНрд░рд╛рд░рдВрдн YOLOv8 рд╡реИрдХрд▓реНрдкрд┐рдХ рдЕрд░реНрдз-рд╕рдЯреАрдХ рд╕рдорд░реНрдерди рдХреЗ рд╕рд╛рде ToTensor рдСрдмреНрдЬреЗрдХреНрдЯред
ultralytics.data.augment.v8_transforms(dataset, imgsz, hyp, stretch=False)
рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рдЙрдкрдпреБрдХреНрдд рдЖрдХрд╛рд░ рдореЗрдВ рдмрджрд▓реЗрдВ YOLOv8 рдкреНрд░рд╢рд┐рдХреНрд╖рдгред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.classify_transforms(size=224, mean=DEFAULT_MEAN, std=DEFAULT_STD, interpolation=Image.BILINEAR, crop_fraction=DEFAULT_CROP_FRACTION)
рд╡рд░реНрдЧреАрдХрд░рдг рдореВрд▓реНрдпрд╛рдВрдХрди/рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдмрджрд▓ рдЬрд╛рддрд╛ рд╣реИред рдЯрд┐рдо/рдбреЗрдЯрд╛/transforms_factory.py рд╕реЗ рдкреНрд░реЗрд░рд┐рддред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
size |
int
|
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ |
224
|
mean |
tuple
|
RGB рдЪреНрдпрд╛рдирд▓рд╣рд░реВрдХреЛ рдФрд╕рдд рдореВрд▓реНрдпрд╣рд░реВ |
DEFAULT_MEAN
|
std |
tuple
|
RGB рдЪреНрдпрд╛рдирд▓рд╣рд░реВрдХреЛ std рдорд╛рдирд╣рд░реВ |
DEFAULT_STD
|
interpolation |
InterpolationMode
|
рдкреНрд░рдХреНрд╖реЗрдк рдореЛрдбред рдбрд┐рдлрд╝реЙрд▓реНрдЯ T.InterpolationMode.BILINEAR рд╣реИред |
BILINEAR
|
crop_fraction |
float
|
рдлрд╕рд▓ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рдХрд╛ рдЕрдВрд╢ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1.0 рд╣реИред |
DEFAULT_CROP_FRACTION
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Compose
|
рдорд╢рд╛рд▓ рджреГрд╖реНрдЯрд┐ рдмрджрд▓ рдЬрд╛рддреА рд╣реИ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
ultralytics.data.augment.classify_augmentations(size=224, mean=DEFAULT_MEAN, std=DEFAULT_STD, scale=None, ratio=None, hflip=0.5, vflip=0.0, auto_augment=None, hsv_h=0.015, hsv_s=0.4, hsv_v=0.4, force_color_jitter=False, erasing=0.0, interpolation=Image.BILINEAR)
рд╡рд░реНрдЧреАрдХрд░рдг рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рд▓рд┐рдП рд╡реГрджреНрдзрд┐ рдХреЗ рд╕рд╛рде рдмрджрд▓ рдЬрд╛рддрд╛ рд╣реИред рдЯрд┐рдо/рдбреЗрдЯрд╛/transforms_factory.py рд╕реЗ рдкреНрд░реЗрд░рд┐рддред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
size |
int
|
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ |
224
|
scale |
tuple
|
рдЫрд╡рд┐ рдХреА рд╕реНрдХреЗрд▓ рд░реЗрдВрдЬред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд╣реИ (0.08, 1.0) |
None
|
ratio |
tuple
|
рдЫрд╡рд┐ рдХреА рдкрд╣рд▓реВ рдЕрдиреБрдкрд╛рдд рд╕реАрдорд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ (3./4., 4./3.) рд╣реИред |
None
|
mean |
tuple
|
RGB рдЪреНрдпрд╛рдирд▓рд╣рд░реВрдХреЛ рдФрд╕рдд рдореВрд▓реНрдпрд╣рд░реВ |
DEFAULT_MEAN
|
std |
tuple
|
RGB рдЪреНрдпрд╛рдирд▓рд╣рд░реВрдХреЛ std рдорд╛рдирд╣рд░реВ |
DEFAULT_STD
|
hflip |
float
|
рдХреНрд╖реИрддрд┐рдЬ рдлреНрд▓рд┐рдк рдХреА рд╕рдВрднрд╛рд╡рдирд╛ |
0.5
|
vflip |
float
|
рд╡рд░реНрдЯрд┐рдХрд▓ рдлреНрд▓рд┐рдк рдХреА рдкреНрд░рд╛рдпрд┐рдХрддрд╛ |
0.0
|
auto_augment |
str
|
рдСрдЯреЛ рд╡реГрджреНрдзрд┐ рдиреАрддрд┐ред 'рд░реИрдВрдбрдЧрдореЗрдВрдЯ', 'рдСрдЧрдорд┐рдХреНрд╕', 'рдСрдЯреЛрдСрдЧрдореЗрдВрдЯ' рдпрд╛ рдХреЛрдИ рдирд╣реАрдВ рд╣реЛ рд╕рдХрддрд╛ рд╣реИред |
None
|
hsv_h |
float
|
рдЫрд╡рд┐ рдПрдЪрдПрд╕рд╡реА-рд╣реНрдпреВ рд╡реГрджреНрдзрд┐ (рдЕрдВрд╢) |
0.015
|
hsv_s |
float
|
рдЫрд╡рд┐ рдПрдЪрдПрд╕рд╡реА-рд╕рдВрддреГрдкреНрддрд┐ рд╡реГрджреНрдзрд┐ (рдЕрдВрд╢) |
0.4
|
hsv_v |
float
|
рдЫрд╡рд┐ рдПрдЪрдПрд╕рд╡реА-рдореВрд▓реНрдп рд╡реГрджреНрдзрд┐ (рдЕрдВрд╢) |
0.4
|
force_color_jitter |
bool
|
рдСрдЯреЛ рд╡реГрджреНрдзрд┐ рд╕рдХреНрд╖рдо рд╣реЛрдиреЗ рдкрд░ рднреА рд░рдВрдЧ рдЬрд┐рдЯрд░ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдмрд▓ |
False
|
erasing |
float
|
рдпрд╛рджреГрдЪреНрдЫрд┐рдХ рдорд┐рдЯрд╛рдиреЗ рдХреА рд╕рдВрднрд╛рд╡рдирд╛ |
0.0
|
interpolation |
InterpolationMode
|
рдкреНрд░рдХреНрд╖реЗрдк рдореЛрдбред рдбрд┐рдлрд╝реЙрд▓реНрдЯ T.InterpolationMode.BILINEAR рд╣реИред |
BILINEAR
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Compose
|
рдорд╢рд╛рд▓ рджреГрд╖реНрдЯрд┐ рдмрджрд▓ рдЬрд╛рддреА рд╣реИ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/augment.py
1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 |
|
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (4)