image_size: W, H
(self, image_size)
| 362 | return strides |
| 363 | |
| 364 | def update(self, image_size): |
| 365 | """ |
| 366 | image_size: W, H |
| 367 | """ |
| 368 | if self.image_size == image_size: |
| 369 | return |
| 370 | anchor_grid, scaler = generate_anchors(image_size, self.strides) |
| 371 | self.image_size = image_size |
| 372 | self.anchor_grid, self.scaler = anchor_grid.to(self.device), scaler.to(self.device) |
| 373 | |
| 374 | def __call__(self, predicts): |
| 375 | preds_cls, preds_anc, preds_box = [], [], [] |