рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/models/sam/predict.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/рдореЙрдбрд▓/sam/predict.pyред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.models.sam.predict.Predictor
рдХрд╛ рд░реВрдк: BasePredictor
рд╕реЗрдЧрдореЗрдВрдЯ рдПрдиреАрдерд┐рдВрдЧ рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рдкреНрд░реЗрдбрд┐рдХреНрдЯрд░ рдХреНрд▓рд╛рд╕ (SAM), BasePredictor рдХрд╛ рд╡рд┐рд╕реНрддрд╛рд░ред
рд╡рд░реНрдЧ рдЫрд╡рд┐ рд╡рд┐рднрд╛рдЬрди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рдЕрдиреБрд░реВрдк рдореЙрдбрд▓ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдПрдХ рдЗрдВрдЯрд░рдлрд╝реЗрд╕ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред рдЙрдиреНрдирдд рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рдФрд░ рд╢реАрдШреНрд░ рд╡рд┐рднрд╛рдЬрди рдХреНрд╖рдорддрд╛рдУрдВ рдХреЗ рд╕рд╛рде, рдпрд╣ рд▓рдЪреАрд▓реЗ рдФрд░ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдХреА рд╕реБрд╡рд┐рдзрд╛ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ рдореБрдЦреМрдЯрд╛ рдкреАрдврд╝реАред рдХрдХреНрд╖рд╛ рд╡рд┐рднрд┐рдиреНрди рдкреНрд░рдХрд╛рд░ рдХреЗ рд╕рдВрдХреЗрддреЛрдВ рдЬреИрд╕реЗ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЗ рд╕рд╛рде рдХрд╛рдо рдХрд░рдиреЗ рдореЗрдВ рд╕рдХреНрд╖рдо рд╣реИ, рдЕрдВрдХ, рдФрд░ рдХрдо-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рдорд╛рд╕реНрдХред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
cfg |
dict
|
рдореЙрдбрд▓ рдФрд░ рдХрд╛рд░реНрдп-рд╕рдВрдмрдВрдзрд┐рдд рдорд╛рдкрджрдВрдбреЛрдВ рдХреЛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдХрд░рдиреЗ рд╡рд╛рд▓рд╛ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рд╢рдмреНрджрдХреЛрд╢ред |
overrides |
dict
|
рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдХреЛ рдУрд╡рд░рд░рд╛рдЗрдб рдХрд░рдиреЗ рд╡рд╛рд▓реЗ рдорд╛рдиреЛрдВ рд╡рд╛рд▓реЗ рд╢рдмреНрджрдХреЛрд╢. |
_callbacks |
dict
|
рд╡реНрдпрд╡рд╣рд╛рд░ рдХреЛ рдмрдврд╝рд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧрдХрд░реНрддрд╛-рдкрд░рд┐рднрд╛рд╖рд┐рдд рдХреЙрд▓рдмреИрдХ рдлрд╝рдВрдХреНрд╢рдВрд╕ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ред |
args |
namespace
|
рдХрдорд╛рдВрдб-рд▓рд╛рдЗрди рддрд░реНрдХ рдпрд╛ рдЕрдиреНрдп рдкрд░рд┐рдЪрд╛рд▓рди рдЪрд░ рд░рдЦрдиреЗ рдХреЗ рд▓рд┐рдП рдирд╛рдорд╕реНрдерд╛рдиред |
im |
Tensor
|
рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕реНрдб рдЗрдирдкреБрдЯ рдЫрд╡рд┐ tensor. |
features |
Tensor
|
рдирд┐рд╖реНрдХрд░реНрд╖ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдирд┐рдХрд╛рд▓реА рдЧрдИ рдЫрд╡рд┐ рд╕реБрд╡рд┐рдзрд╛рдПрдБред |
prompts |
dict
|
рд╡рд┐рднрд┐рдиреНрди рд╢реАрдШреНрд░ рдкреНрд░рдХрд╛рд░реЛрдВ рдХрд╛ рд╕рдВрдЧреНрд░рд╣, рдЬреИрд╕реЗ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рдкреЙрдЗрдВрдЯреНрд╕ред |
segment_all |
bool
|
рдЫрд╡рд┐ рдореЗрдВ рд╕рднреА рдСрдмреНрдЬреЗрдХреНрдЯреНрд╕ рдХреЛ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдпрд╛ рдХреЗрд╡рд▓ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рд▓реЛрдЧреЛрдВ рдХреЛ рдирд┐рдпрдВрддреНрд░рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдзреНрд╡рдЬред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
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 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 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 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 |
|
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
рдкреНрд░реЗрдбрд┐рдХреНрдЯрд░ рдХреЛ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди, рдУрд╡рд░рд░рд╛рдЗрдб рдФрд░ рдХреЙрд▓рдмреИрдХ рдХреЗ рд╕рд╛рде рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВред
рд╡рд┐рдзрд┐ рдкреНрд░реЗрдбрд┐рдХреНрдЯрд░ рдСрдмреНрдЬреЗрдХреНрдЯ рд╕реЗрдЯ рдХрд░рддреА рд╣реИ рдФрд░ рдкреНрд░рджрд╛рди рдХрд┐рдП рдЧрдП рдХрд┐рд╕реА рднреА рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдУрд╡рд░рд░рд╛рдЗрдб рдпрд╛ рдХреЙрд▓рдмреИрдХ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рддреА рд╣реИред рдпрд╣ рдХреЗ рд▓рд┐рдП рдХрд╛рд░реНрдп-рд╡рд┐рд╢рд┐рд╖реНрдЯ рд╕реЗрдЯрд┐рдВрдЧреНрд╕ рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИ SAM, рдЬреИрд╕реЗ рдХрд┐ рдЗрд╖реНрдЯрддрдо рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЗ рд▓рд┐рдП retina_masks True рдкрд░ рд╕реЗрдЯ рдХрд┐рдпрд╛ рдЬрд╛ рд░рд╣рд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
cfg |
dict
|
рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рд╢рдмреНрджрдХреЛрд╢ред |
DEFAULT_CFG
|
overrides |
dict
|
рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдХреЛ рдУрд╡рд░рд░рд╛рдЗрдб рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдорд╛рдиреЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ред |
None
|
_callbacks |
dict
|
рд╡реНрдпрд╡рд╣рд╛рд░ рдХреЛ рдЕрдиреБрдХреВрд▓рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдХреЙрд▓рдмреИрдХ рдлрд╝рдВрдХреНрд╢рдВрд╕ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ред |
None
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
generate(im, crop_n_layers=0, crop_overlap_ratio=512 / 1500, crop_downscale_factor=1, point_grids=None, points_stride=32, points_batch_size=64, conf_thres=0.88, stability_score_thresh=0.95, stability_score_offset=0.95, crop_nms_thresh=0.7)
рд╕реЗрдЧрдореЗрдВрдЯ рдПрдиреАрдерд┐рдВрдЧ рдореЙрдбрд▓ (SAM).
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рд▓реАрд╡рд░реЗрдЬ рджреНрд╡рд╛рд░рд╛ рдПрдХ рд╕рдВрдкреВрд░реНрдг рдЫрд╡рд┐ рдХреЛ рдШрдЯрдХ рднрд╛рдЧреЛрдВ рдореЗрдВ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд░рддрд╛ рд╣реИ SAMрдЙрдиреНрдирдд рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рдФрд░ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдкреНрд░рджрд░реНрд╢рди рдХреНрд╖рдорддрд╛рдУрдВред рдпрд╣ рд╡реИрдХрд▓реНрдкрд┐рдХ рд░реВрдк рд╕реЗ рдмреЗрд╣рддрд░ рд╡рд┐рднрд╛рдЬрди рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рдлрд╕рд▓реЛрдВ рдкрд░ рдХрд╛рдо рдХрд░ рд╕рдХрддреЗ рд╣реИрдВ.
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
Tensor
|
рдирд┐рд╡реЗрд╢ tensor рдЖрдпрд╛рдореЛрдВ (рдПрди, рд╕реА, рдПрдЪ, рдбрдмреНрд▓реНрдпреВ) рдХреЗ рд╕рд╛рде рдкреВрд░реНрд╡рд╕рдВрд╕рд╛рдзрд┐рдд рдЫрд╡рд┐ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рдирд╛ред |
рдЖрд╡рд╢реНрдпрдХ |
crop_n_layers |
int
|
рдЫрд╡рд┐ рдлрд╕рд▓реЛрдВ рдкрд░ рдЕрддрд┐рд░рд┐рдХреНрдд рдореБрдЦреМрдЯрд╛ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рд▓рд┐рдП рдкрд░рддреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдХрд░рддрд╛ рд╣реИред рдкреНрд░рддреНрдпреЗрдХ рдкрд░рдд 2 ** i_layer рдЫрд╡рд┐ рдлрд╕рд▓реЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рдХрд╛ рдЙрддреНрдкрд╛рджрди рдХрд░рддреА рд╣реИред |
0
|
crop_overlap_ratio |
float
|
рдлрд╕рд▓реЛрдВ рдХреЗ рдмреАрдЪ рдУрд╡рд░рд▓реИрдк рдирд┐рд░реНрдзрд╛рд░рд┐рдд рдХрд░рддрд╛ рд╣реИред рдмрд╛рдж рдХреА рдкрд░рддреЛрдВ рдореЗрдВ рдиреАрдЪреЗ рдЧрд┐рд░рд╛рдпрд╛ рдЧрдпрд╛ред |
512 / 1500
|
crop_downscale_factor |
int
|
рдкреНрд░рддреНрдпреЗрдХ рдкрд░рдд рдореЗрдВ рдирдореВрдирд╛ рдЕрдВрдХ рдкреНрд░рддрд┐ рдкрдХреНрд╖ рдХреА рд╕рдВрдЦреНрдпрд╛ рдХреЗ рд▓рд┐рдП рд╕реНрдХреЗрд▓рд┐рдВрдЧ рдХрд╛рд░рдХ. |
1
|
point_grids |
list[ndarray]
|
рдмрд┐рдВрджреБ рдирдореВрдирд╛рдХрд░рдг рдХреЗ рд▓рд┐рдП рдХрд╕реНрдЯрдо рдЧреНрд░рд┐рдб [0,1] рдХреЗ рд▓рд┐рдП рд╕рд╛рдорд╛рдиреНрдпреАрдХреГрдд рд╣реИрдВред nth рдлрд╕рд▓ рдкрд░рдд рдореЗрдВ рдЙрдкрдпреЛрдЧ рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИред |
None
|
points_stride |
int
|
рдЫрд╡рд┐ рдХреЗ рдкреНрд░рддреНрдпреЗрдХ рдкрдХреНрд╖ рдХреЗ рд╕рд╛рде рдирдореВрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЕрдВрдХ рдХреА рд╕рдВрдЦреНрдпрд╛. 'point_grids' рдХреЗ рд╕рд╛рде рд╡рд┐рд╢реЗрд╖ред |
32
|
points_batch_size |
int
|
рдПрдХ рд╕рд╛рде рд╕рдВрд╕рд╛рдзрд┐рдд рдЕрдВрдХреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рдХреЗ рд▓рд┐рдП рдмреИрдЪ рдЖрдХрд╛рд░ред |
64
|
conf_thres |
float
|
рдореЙрдбрд▓ рдХреА рдореБрдЦреМрдЯрд╛ рдЧреБрдгрд╡рддреНрддрд╛ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдлрд╝рд┐рд▓реНрдЯрд░рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реАрдорд╛ [0,1]ред |
0.88
|
stability_score_thresh |
float
|
рд╕реНрдерд┐рд░рддрд╛ рд╕реАрдорд╛ [0,1] рдорд╛рд╕реНрдХ рд╕реНрдерд┐рд░рддрд╛ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдорд╛рд╕реНрдХ рдлрд╝рд┐рд▓реНрдЯрд░рд┐рдВрдЧ рдХреЗ рд▓рд┐рдПред |
0.95
|
stability_score_offset |
float
|
рд╕реНрдерд┐рд░рддрд╛ рд╕реНрдХреЛрд░ рдХреА рдЧрдгрдирд╛ рдХреЗ рд▓рд┐рдП рдСрдлрд╕реЗрдЯ рдореВрд▓реНрдпред |
0.95
|
crop_nms_thresh |
float
|
рдлрд╕рд▓реЛрдВ рдХреЗ рдмреАрдЪ рдбреБрдкреНрд▓рд┐рдХреЗрдЯ рдорд╛рд╕реНрдХ рд╣рдЯрд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдирдПрдордПрд╕ рдХреЗ рд▓рд┐рдП IoU рдХрдЯрдСрдлред |
0.7
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple
|
рдПрдХ рдЯрдкрд▓ рдЬрд┐рд╕рдореЗрдВ рдЦрдВрдбрд┐рдд рдорд╛рд╕реНрдХ, рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реНрдХреЛрд░ рдФрд░ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рд╣реЛрддреЗ рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
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 |
|
inference(im, bboxes=None, points=None, labels=None, masks=None, multimask_output=False, *args, **kwargs)
рд╡рд░реНрддрдорд╛рди рдореЗрдВ рднрд░реА рд╣реБрдИ рдЫрд╡рд┐ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ, рджрд┐рдП рдЧрдП рдЗрдирдкреБрдЯ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдЫрд╡рд┐ рд╡рд┐рднрд╛рдЬрди рдЕрдиреБрдорд╛рди рдХрд░реЗрдВред рдпрд╣рди рд╡рд┐рдзрд┐ рдХрд╛ рд▓рд╛рдн рдЙрдард╛рддрд╛ рд╣реИ SAM(рд╕реЗрдЧрдореЗрдВрдЯ рдПрдиреАрдерд┐рдВрдЧ рдореЙрдбрд▓) рдЖрд░реНрдХрд┐рдЯреЗрдХреНрдЪрд░ рдЬрд┐рд╕рдореЗрдВ рдЗрдореЗрдЬ рдПрдирдХреЛрдбрд░, рдкреНрд░реЙрдореНрдкреНрдЯ рдПрдирдХреЛрдбрд░ рдФрд░ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдФрд░ рд╢реАрдШреНрд░ рд╡рд┐рднрд╛рдЬрди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдорд╛рд╕реНрдХ рдбрд┐рдХреЛрдбрд░ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
Tensor
|
рдореЗрдВ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕реНрдб рдЗрдирдкреБрдЯ рдЫрд╡рд┐ tensor рдкреНрд░рд╛рд░реВрдк, рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде (рдПрди, рд╕реА, рдПрдЪ, рдбрдмреНрд▓реНрдпреВ)ред |
рдЖрд╡рд╢реНрдпрдХ |
bboxes |
ndarray | List
|
рдЖрдХрд╛рд░ (N, 4) рдХреЗ рд╕рд╛рде рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, XYXY рдкреНрд░рд╛рд░реВрдк рдореЗрдВред |
None
|
points |
ndarray | List
|
рдкрд┐рдХреНрд╕реЗрд▓ рдореЗрдВ рдЖрдХреГрддрд┐ (N, 2) рдХреЗ рд╕рд╛рде рдСрдмреНрдЬреЗрдХреНрдЯ рд╕реНрдерд╛рдиреЛрдВ рдХреЛ рдЗрдВрдЧрд┐рдд рдХрд░рдиреЗ рд╡рд╛рд▓реЗ рдмрд┐рдВрджреБ. |
None
|
labels |
ndarray | List
|
рдмрд┐рдВрджреБ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рд▓рд┐рдП рд▓реЗрдмрд▓, рдЖрдХрд╛рд░ (рдПрди, )ред 1 = рдЕрдЧреНрд░рднреВрдорд┐, 0 = рдкреГрд╖реНрдарднреВрдорд┐ред |
None
|
masks |
ndarray
|
рдкрд┐рдЫрд▓реЗ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рдЖрдХрд╛рд░ рд╕реЗ рдХрдо-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рдорд╛рд╕реНрдХ (рдПрди, рдПрдЪ, рдбрдмреНрд▓реНрдпреВ)ред рдХреЗ рд▓рд┐рдП SAM рдПрдЪ = рдбрдмреНрд▓реНрдпреВ = 256ред |
None
|
multimask_output |
bool
|
рдПрдХрд╛рдзрд┐рдХ рдорд╛рд╕реНрдХ рд╡рд╛рдкрд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдзреНрд╡рдЬред рдЕрд╕реНрдкрд╖реНрдЯ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рд▓рд┐рдП рд╕рд╣рд╛рдпрдХред |
False
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple
|
рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рддреАрди рддрддреНрд╡ рд╢рд╛рдорд┐рд▓ рд╣реИрдВред - np.ndarray: рдЖрдХрд╛рд░ CxHxW рдореЗрдВ рдЖрдЙрдЯрдкреБрдЯ рдорд╛рд╕реНрдХ, рдЬрд╣рд╛рдВ C рдЙрддреНрдкрдиреНрди рдорд╛рд╕реНрдХ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИред - np.ndarray: рд▓рдВрдмрд╛рдИ рд╕реА рдХреА рдПрдХ рд╕рд░рдгреА рдЬрд┐рд╕рдореЗрдВ рдкреНрд░рддреНрдпреЗрдХ рдорд╛рд╕реНрдХ рдХреЗ рд▓рд┐рдП рдореЙрдбрд▓ рджреНрд╡рд╛рд░рд╛ рдЕрдиреБрдорд╛рдирд┐рдд рдЧреБрдгрд╡рддреНрддрд╛ рд╕реНрдХреЛрд░ рд╣реЛрддреЗ рд╣реИрдВред - np.ndarray: рдмрд╛рдж рдХреЗ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЖрдХрд╛рд░ CxHxW рдХреЗ рдХрдо-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рд▓реЙрдЧ, рдЬрд╣рд╛рдВ H=W=256ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
postprocess(preds, img, orig_imgs)
рдкреЛрд╕реНрдЯ-рдкреНрд░рдХреНрд░рд┐рдпрд╛рдПрдВ SAMрдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдорд╛рд╕реНрдХ рдФрд░ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдЙрддреНрдкрдиреНрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЕрдиреБрдорд╛рди рдЖрдЙрдЯрдкреБрдЯ рдХрд░рддрд╛ рд╣реИред
рд╡рд┐рдзрд┐ рдорд╛рд╕реНрдХ рдФрд░ рдмрдХреНрд╕реЗ рдХреЛ рдореВрд▓ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдореЗрдВ рдорд╛рдкрддреА рд╣реИ рдФрд░ рдореБрдЦреМрдЯрд╛ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рд▓рд┐рдП рдПрдХ рд╕реАрдорд╛ рд▓рд╛рдЧреВ рдХрд░рддреА рд╣реИред рд╡рд╣реА SAM рдореЙрдбрд▓ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдХреЛ рдкреНрд░рд╛рдкреНрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдиреНрдирдд рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рдФрд░ рд╢реАрдШреНрд░ рд╡рд┐рднрд╛рдЬрди рдХрд╛рд░реНрдпреЛрдВ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
preds |
tuple
|
рд╕реЗ рдЖрдЙрдЯрдкреБрдЯ SAM рдореЙрдбрд▓ рдЕрдиреБрдорд╛рди, рдЬрд┐рд╕рдореЗрдВ рдорд╛рд╕реНрдХ, рд╕реНрдХреЛрд░ рдФрд░ рд╡реИрдХрд▓реНрдкрд┐рдХ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рд╢рд╛рдорд┐рд▓ рд╣реИрдВред |
рдЖрд╡рд╢реНрдпрдХ |
img |
Tensor
|
рд╕рдВрд╕рд╛рдзрд┐рдд рдЗрдирдкреБрдЯ рдЫрд╡рд┐ tensor. |
рдЖрд╡рд╢реНрдпрдХ |
orig_imgs |
list | Tensor
|
рдореВрд▓, рдЕрд╕рдВрд╕рд╛рдзрд┐рдд рдЪрд┐рддреНрд░ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
list
|
рдбрд┐рдЯреЗрдХреНрд╢рди рдорд╛рд╕реНрдХ, рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рдЕрдиреНрдп рдореЗрдЯрд╛рдбреЗрдЯрд╛ рд╡рд╛рд▓реЗ рдкрд░рд┐рдгрд╛рдо рдСрдмреНрдЬреЗрдХреНрдЯреНрд╕ рдХреА рд╕реВрдЪреАред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
pre_transform(im)
рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдЗрдирдкреБрдЯ рдЗрдореЗрдЬ рдкрд░ рдкреНрд░рд╛рд░рдВрднрд┐рдХ рдкрд░рд┐рд╡рд░реНрддрди рдХрд░реЗрдВред
рд╡рд┐рдзрд┐ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдХреЛ рд▓рд╛рдЧреВ рдХрд░рддреА рд╣реИ рдЬреИрд╕реЗ рдХрд┐ рдЖрдЧреЗ рдХреЗ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рддреИрдпрд╛рд░ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЖрдХрд╛рд░ рдмрджрд▓рдирд╛ред рд╡рд░реНрддрдорд╛рди рдореЗрдВ, рдмреИрдЪ рдХрд┐рдП рдЧрдП рдЕрдиреБрдорд╛рди рд╕рдорд░реНрдерд┐рдд рдирд╣реАрдВ рд╣реИрдВ; рдЗрд╕рд▓рд┐рдП рд╕реВрдЪреА рдХреА рд▓рдВрдмрд╛рдИ 1 рд╣реЛрдиреА рдЪрд╛рд╣рд┐рдПред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
List[ndarray]
|
HWC numpy рд╕рд░рдгреА рдкреНрд░рд╛рд░реВрдк рдореЗрдВ рдЫрд╡рд┐рдпреЛрдВ рд╡рд╛рд▓реА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
List[ndarray]
|
рд░реВрдкрд╛рдВрддрд░рд┐рдд рдЫрд╡рд┐рдпреЛрдВ рдХреА рд╕реВрдЪреАред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
preprocess(im)
рдореЙрдбрд▓ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЗрдирдкреБрдЯ рдЫрд╡рд┐ рдХреЛ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕ рдХрд░реЗрдВред
рд╡рд┐рдзрд┐ рдкрд░рд┐рд╡рд░реНрддрдиреЛрдВ рдФрд░ рд╕рд╛рдорд╛рдиреНрдпреАрдХрд░рдг рдХреЛ рд▓рд╛рдЧреВ рдХрд░рдХреЗ рдЗрдирдкреБрдЯ рдЫрд╡рд┐ рддреИрдпрд╛рд░ рдХрд░рддреА рд╣реИред рдпрд╣ рджреЛрдиреЛрдВ рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддрд╛ рд╣реИ torch.Tensor рдФрд░ рдЗрдирдкреБрдЯ рдкреНрд░рд╛рд░реВрдкреЛрдВ рдХреЗ рд░реВрдк рдореЗрдВ np.ndarray рдХреА рд╕реВрдЪреАред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
Tensor | List[ndarray]
|
рдмреАрд╕реАрдПрдЪрдбрдмреНрд▓реНрдпреВ tensor рдкреНрд░рд╛рд░реВрдк рдпрд╛ HWC numpy рд╕рд░рдгрд┐рдпреЛрдВ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдкреВрд░реНрд╡рд╕рдВрд╕рд╛рдзрд┐рдд рдЫрд╡рд┐ tensor. |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
prompt_inference(im, bboxes=None, points=None, labels=None, masks=None, multimask_output=False)
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, рдкреЙрдЗрдВрдЯреНрд╕ рдФрд░ рдорд╛рд╕реНрдХ рдЬреИрд╕реЗ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдЫрд╡рд┐ рд╡рд┐рднрд╛рдЬрди рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЖрдВрддрд░рд┐рдХ рдХрд╛рд░реНрдпред рдлрд╝рд╛рдпрджрд╛ рдЙрдард╛рдирд╛ SAMрд╢реАрдШреНрд░-рдЖрдзрд╛рд░рд┐рдд, рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рд╡рд┐рднрд╛рдЬрди рдХреЗ рд▓рд┐рдП рд╡рд┐рд╢реЗрд╖ рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im |
Tensor
|
рдореЗрдВ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕реНрдб рдЗрдирдкреБрдЯ рдЫрд╡рд┐ tensor рдкреНрд░рд╛рд░реВрдк, рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде (рдПрди, рд╕реА, рдПрдЪ, рдбрдмреНрд▓реНрдпреВ)ред |
рдЖрд╡рд╢реНрдпрдХ |
bboxes |
ndarray | List
|
рдЖрдХрд╛рд░ (N, 4) рдХреЗ рд╕рд╛рде рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, XYXY рдкреНрд░рд╛рд░реВрдк рдореЗрдВред |
None
|
points |
ndarray | List
|
рдкрд┐рдХреНрд╕реЗрд▓ рдореЗрдВ рдЖрдХреГрддрд┐ (N, 2) рдХреЗ рд╕рд╛рде рдСрдмреНрдЬреЗрдХреНрдЯ рд╕реНрдерд╛рдиреЛрдВ рдХреЛ рдЗрдВрдЧрд┐рдд рдХрд░рдиреЗ рд╡рд╛рд▓реЗ рдмрд┐рдВрджреБ. |
None
|
labels |
ndarray | List
|
рдмрд┐рдВрджреБ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рд▓рд┐рдП рд▓реЗрдмрд▓, рдЖрдХрд╛рд░ (рдПрди, )ред 1 = рдЕрдЧреНрд░рднреВрдорд┐, 0 = рдкреГрд╖реНрдарднреВрдорд┐ред |
None
|
masks |
ndarray
|
рдкрд┐рдЫрд▓реЗ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рдЖрдХрд╛рд░ рд╕реЗ рдХрдо-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рдорд╛рд╕реНрдХ (рдПрди, рдПрдЪ, рдбрдмреНрд▓реНрдпреВ)ред рдХреЗ рд▓рд┐рдП SAM рдПрдЪ = рдбрдмреНрд▓реНрдпреВ = 256ред |
None
|
multimask_output |
bool
|
рдПрдХрд╛рдзрд┐рдХ рдорд╛рд╕реНрдХ рд╡рд╛рдкрд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдзреНрд╡рдЬред рдЕрд╕реНрдкрд╖реНрдЯ рд╕рдВрдХреЗрддреЛрдВ рдХреЗ рд▓рд┐рдП рд╕рд╣рд╛рдпрдХред |
False
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple
|
рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рддреАрди рддрддреНрд╡ рд╢рд╛рдорд┐рд▓ рд╣реИрдВред - np.ndarray: рдЖрдХрд╛рд░ CxHxW рдореЗрдВ рдЖрдЙрдЯрдкреБрдЯ рдорд╛рд╕реНрдХ, рдЬрд╣рд╛рдВ C рдЙрддреНрдкрдиреНрди рдорд╛рд╕реНрдХ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИред - np.ndarray: рд▓рдВрдмрд╛рдИ рд╕реА рдХреА рдПрдХ рд╕рд░рдгреА рдЬрд┐рд╕рдореЗрдВ рдкреНрд░рддреНрдпреЗрдХ рдорд╛рд╕реНрдХ рдХреЗ рд▓рд┐рдП рдореЙрдбрд▓ рджреНрд╡рд╛рд░рд╛ рдЕрдиреБрдорд╛рдирд┐рдд рдЧреБрдгрд╡рддреНрддрд╛ рд╕реНрдХреЛрд░ рд╣реЛрддреЗ рд╣реИрдВред - np.ndarray: рдмрд╛рдж рдХреЗ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЖрдХрд╛рд░ CxHxW рдХреЗ рдХрдо-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рд▓реЙрдЧ, рдЬрд╣рд╛рдВ H=W=256ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
remove_small_regions(masks, min_area=0, nms_thresh=0.7)
staticmethod
рд╕реЗрдЧрдореЗрдВрдЯ рдПрдиреАрдерд┐рдВрдЧ рдореЙрдбрд▓ рджреНрд╡рд╛рд░рд╛ рдЙрддреНрдкрдиреНрди рд╕реЗрдЧрдореЗрдВрдЯреЗрд╢рди рдорд╛рд╕реНрдХ рдкрд░ рдкреЛрд╕реНрдЯ-рдкреНрд░реЛрд╕реЗрд╕рд┐рдВрдЧ рдХрд░реЗрдВ (SAM). рд╡рд┐рд╢реЗрд╖ рд░реВрдк рд╕реЗ, рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдЗрдирдкреБрдЯ рдорд╛рд╕реНрдХ рд╕реЗ рдЫреЛрдЯреЗ рдбрд┐рд╕реНрдХрдиреЗрдХреНрдЯ рдХрд┐рдП рдЧрдП рдХреНрд╖реЗрддреНрд░реЛрдВ рдФрд░ рдЫреЗрджреЛрдВ рдХреЛ рд╣рдЯрд╛ рджреЗрддрд╛ рд╣реИ, рдФрд░ рдлрд┐рд░ рдЧреИрд░-рдЕрдзрд┐рдХрддрдо рдкреНрд░рджрд░реНрд╢рди рдХрд░рддрд╛ рд╣реИ рджрдорди (рдПрдирдПрдордПрд╕) рдХрд┐рд╕реА рднреА рдирдП рдмрдирд╛рдП рдЧрдП рдбреБрдкреНрд▓рд┐рдХреЗрдЯ рдмреЙрдХреНрд╕ рдХреЛ рдЦрддреНрдо рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдПред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
masks |
Tensor
|
рдПрдХ tensor рдЬрд┐рд╕рдореЗрдВ рдорд╛рд╕реНрдХ рдХреЛ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд┐рдпрд╛ рдЬрд╛рдирд╛ рд╣реИред рдЖрдХрд╛рд░ (N, H, W) рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдП, рдЬрд╣рд╛рдБ N рд╣реИ рдорд╛рд╕реНрдХ рдХреА рд╕рдВрдЦреНрдпрд╛, рдПрдЪ рдКрдВрдЪрд╛рдИ рд╣реИ, рдФрд░ рдбрдмреНрд▓реНрдпреВ рдЪреМрдбрд╝рд╛рдИ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
min_area |
int
|
рдиреНрдпреВрдирддрдо рдХреНрд╖реЗрддреНрд░ рдЬрд┐рд╕рдХреЗ рдиреАрдЪреЗ рдбрд┐рд╕реНрдХрдиреЗрдХреНрдЯ рдХрд┐рдП рдЧрдП рдХреНрд╖реЗрддреНрд░реЛрдВ рдФрд░ рдЫреЗрджреЛрдВ рдХреЛ рд╣рдЯрд╛ рджрд┐рдпрд╛ рдЬрд╛рдПрдЧрд╛ред 0 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
0
|
nms_thresh |
float
|
NMS рдПрд▓реНрдЧреЛрд░рд┐рдердо рдХреЗ рд▓рд┐рдП IoU рдереНрд░реЗрд╢реЛрд▓реНрдбред 0.7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
0.7
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple([Tensor, List[int]])
|
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
reset_image()
рдЫрд╡рд┐ рдФрд░ рдЙрд╕рдХреА рд╡рд┐рд╢реЗрд╖рддрд╛рдУрдВ рдХреЛ рдХреЛрдИ рдирд╣реАрдВ рдкрд░ рд░реАрд╕реЗрдЯ рдХрд░рддрд╛ рд╣реИред
set_image(image)
рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдПрдХрд▓ рдЫрд╡рд┐ рдХреЛ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕ рдФрд░ рд╕реЗрдЯ рдХрд░рддрд╛ рд╣реИред
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдореЙрдбрд▓ рд╕реЗрдЯ рдХрд░рддрд╛ рд╣реИ рдпрджрд┐ рдкрд╣рд▓реЗ рд╕реЗ рд╣реА рдкреНрд░рд╛рд░рдВрдн рдирд╣реАрдВ рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИ, рддреЛ рдбреЗрдЯрд╛ рд╕реНрд░реЛрдд рдХреЛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЫрд╡рд┐ рдореЗрдВ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░ рдХрд░рддрд╛ рд╣реИ, рдФрд░ рдлреАрдЪрд░ рдирд┐рд╖реНрдХрд░реНрд╖рдг рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рдХреЛ рдкреНрд░реАрдкреНрд░реЛрд╕реЗрд╕ рдХрд░рддрд╛ рд╣реИред рдПрдХ рд╕рдордп рдореЗрдВ рдХреЗрд╡рд▓ рдПрдХ рдЫрд╡рд┐ рд╕реЗрдЯ рдХреА рдЬрд╛ рд╕рдХрддреА рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
image |
str | ndarray
|
рд╕реНрдЯреНрд░рд┐рдВрдЧ рдХреЗ рд░реВрдк рдореЗрдВ рдЫрд╡рд┐ рдлрд╝рд╛рдЗрд▓ рдкрде, рдпрд╛ cv2 рджреНрд╡рд╛рд░рд╛ рдкрдврд╝реА рдЧрдИ np.ndarray рдЫрд╡рд┐ред |
рдЖрд╡рд╢реНрдпрдХ |
рдЙрдард╛рддреА:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
AssertionError
|
рдпрджрд┐ рдПрдХ рд╕реЗ рдЕрдзрд┐рдХ рдЫрд╡рд┐ рд╕реЗрдЯ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
set_prompts(prompts)
setup_model(model, verbose=True)
рд╕реЗрдЧрдореЗрдВрдЯ рдПрдиреАрдерд┐рдВрдЧ рдореЙрдбрд▓ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИ (SAM) рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдПред
рдпрд╣ рд╡рд┐рдзрд┐ рд╕реЗрдЯ рдХрд░рддреА рд╣реИ SAM рдореЙрдбрд▓ рдХреЛ рдЙрдкрдпреБрдХреНрдд рдбрд┐рд╡рд╛рдЗрд╕ рдкрд░ рдЖрд╡рдВрдЯрд┐рдд рдХрд░рдХреЗ рдФрд░ рдЖрд╡рд╢реНрдпрдХ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рдХреЗ рдЫрд╡рд┐ рд╕рд╛рдорд╛рдиреНрдпреАрдХрд░рдг рдФрд░ рдЕрдиреНрдп рдХреЗ рд▓рд┐рдП рдкреИрд░рд╛рдореАрдЯрд░ Ultralytics рд╕рдВрдЧрддрддрд╛ рд╕реЗрдЯрд┐рдВрдЧреНрд╕ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
model |
Module
|
рдПрдХ рдкреВрд░реНрд╡ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд SAM рдХреЛ рдЧрдврд╝рдирд╛ред рдпрджрд┐ рдХреЛрдИ рдирд╣реАрдВ, рддреЛ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдПрдХ рдореЙрдбрд▓ рдмрдирд╛рдпрд╛ рдЬрд╛рдПрдЧрд╛ред |
рдЖрд╡рд╢реНрдпрдХ |
verbose |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдЪрдпрдирд┐рдд рдбрд┐рд╡рд╛рдЗрд╕ рдЬрд╛рдирдХрд╛рд░реА рдкреНрд░рд┐рдВрдЯ рдХрд░рддрд╛ рд╣реИред |
True
|
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
model |
Module
|
рд╡рд╣реА SAM рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЪреБрдиреЗ рдЧрдП рдбрд┐рд╡рд╛рдЗрд╕ рдХреЛ рдЖрд╡рдВрдЯрд┐рдд рдореЙрдбрд▓ред |
device |
device
|
рд╡рд╣ рдЙрдкрдХрд░рдг рдЬрд┐рд╕рдХреЗ рд▓рд┐рдП рдореЙрдбрд▓ рдФрд░ рдЯреЗрдВрд╕рд░ рдЖрд╡рдВрдЯрд┐рдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред |
mean |
Tensor
|
рдЫрд╡рд┐ рд╕рд╛рдорд╛рдиреНрдпреАрдХрд░рдг рдХреЗ рд▓рд┐рдП рдФрд╕рдд рдореВрд▓реНрдпред |
std |
Tensor
|
рдЫрд╡рд┐ рд╕рд╛рдорд╛рдиреНрдпреАрдХрд░рдг рдХреЗ рд▓рд┐рдП рдорд╛рдирдХ рд╡рд┐рдЪрд▓рди рдорд╛рдиред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
setup_source(source)
рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдбреЗрдЯрд╛ рд╕реНрд░реЛрдд рд╕реЗрдЯ рдХрд░рддрд╛ рд╣реИ.
рдпрд╣ рд╡рд┐рдзрд┐ рдЙрд╕ рдбреЗрдЯрд╛ рд╕реНрд░реЛрдд рдХреЛ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░ рдХрд░рддреА рд╣реИ рдЬрд┐рд╕рд╕реЗ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЪрд┐рддреНрд░ рдкреНрд░рд╛рдкреНрдд рдХрд┐рдП рдЬрд╛рдПрдВрдЧреЗред рд╕реНрд░реЛрдд рдПрдХ рд╣реЛ рд╕рдХрддрд╛ рд╣реИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛, рдПрдХ рд╡реАрдбрд┐рдпреЛ рдлрд╝рд╛рдЗрд▓, рдпрд╛ рдЕрдиреНрдп рдкреНрд░рдХрд╛рд░ рдХреЗ рдЫрд╡рд┐ рдбреЗрдЯрд╛ рд╕реНрд░реЛрддред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
source |
str | Path
|
рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рдбреЗрдЯрд╛ рд╕реНрд░реЛрдд рдХрд╛ рдкрде. |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/predict.py
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (3)