kwcoco.demo.toypatterns module¶
-
class
kwcoco.demo.toypatterns.
CategoryPatterns
(categories=None, fg_scale=0.5, fg_intensity=0.9, rng=None)[source]¶ Bases:
object
Example
>>> self = CategoryPatterns.coerce() >>> chip = np.zeros((100, 100, 3)) >>> offset = (20, 10) >>> dims = (160, 140) >>> info = self.random_category(chip, offset, dims) >>> print('info = {}'.format(ub.repr2(info, nl=1))) >>> # xdoctest: +REQUIRES(--show) >>> import kwplot >>> kwplot.autompl() >>> kwplot.imshow(info['data'], pnum=(1, 2, 1), fnum=1, title='chip-space') >>> kpts = kwimage.Points._from_coco(info['keypoints']) >>> kpts.translate(-np.array(offset)).draw(radius=3) >>> ##### >>> mask = kwimage.Mask.coerce(info['segmentation']) >>> kwplot.imshow(mask.to_c_mask().data, pnum=(1, 2, 2), fnum=1, title='img-space') >>> kpts.draw(radius=3) >>> kwplot.show_if_requested()
-
classmethod
coerce
(data=None, **kwargs)[source]¶ Construct category patterns from either defaults or only with specific categories. Can accept either an existig category pattern object, a list of known catnames, or mscoco category dictionaries.
Example
>>> data = ['superstar'] >>> self = CategoryPatterns.coerce(data)
-
random_category
(chip, xy_offset=None, dims=None, newstyle=True)[source]¶ - Ignore:
- import xdev globals().update(xdev.get_func_kwargs(self.random_category))
Example
>>> from kwcoco.demo.toypatterns import * # NOQA >>> self = CategoryPatterns.coerce(['superstar']) >>> chip = np.random.rand(64, 64) >>> info = self.random_category(chip)
-
render_category
(cname, chip, xy_offset=None, dims=None, newstyle=True)[source]¶ - Ignore:
- import xdev globals().update(xdev.get_func_kwargs(self.random_category))
Example
>>> self = CategoryPatterns.coerce(['superstar']) >>> chip = np.random.rand(64, 64) >>> info = self.render_category('superstar', chip, newstyle=True) >>> print('info = {}'.format(ub.repr2(info, nl=-1))) >>> info = self.render_category('superstar', chip, newstyle=False) >>> print('info = {}'.format(ub.repr2(info, nl=-1)))
-
classmethod
-
kwcoco.demo.toypatterns.
star
(a, dtype=<class 'numpy.uint8'>)[source]¶ Generates a star shaped structuring element.
Much faster than skimage.morphology version