This is the initialization for class uniform sampling :param size: crop size (int) :param crop_nopad: Padding or no padding (bool) :param scale_min: Minimum Scale (float) :param scale_max: Maximum Scale (float) :param ignore_index: The index value to
(self, size, crop_nopad, scale_min=0.5, scale_max=2.0, ignore_index=0,
class_list=[16, 15, 14])
| 578 | |
| 579 | class _ClassUniform(object): |
| 580 | def __init__(self, size, crop_nopad, scale_min=0.5, scale_max=2.0, ignore_index=0, |
| 581 | class_list=[16, 15, 14]): |
| 582 | """ |
| 583 | This is the initialization for class uniform sampling |
| 584 | :param size: crop size (int) |
| 585 | :param crop_nopad: Padding or no padding (bool) |
| 586 | :param scale_min: Minimum Scale (float) |
| 587 | :param scale_max: Maximum Scale (float) |
| 588 | :param ignore_index: The index value to ignore in the GT images (unsigned int) |
| 589 | :param class_list: A list of class to sample around, by default Truck, train, bus |
| 590 | """ |
| 591 | self.size = size |
| 592 | self.crop = RandomCrop(self.size, ignore_index=ignore_index, nopad=crop_nopad) |
| 593 | |
| 594 | self.class_list = class_list.replace(" ", "").split(",") |
| 595 | |
| 596 | self.scale_min = scale_min |
| 597 | self.scale_max = scale_max |
| 598 | |
| 599 | def detect_peaks(self, image): |
| 600 | """ |