Spaces:

Leyogho
/

Core-AI-IMAGE

Build error

Core-AI-IMAGE / datasets /visual_sampler /__init__.py

Core

edebe10 almost 2 years ago

409 Bytes

from .sampler import ShapeSampler
from .simpleclick_sampler import SimpleClickSampler


def build_shape_sampler(cfg, **kwargs):
    sampler_name = cfg['STROKE_SAMPLER']['EVAL']['MODE']
    if sampler_name == 'random':
        return ShapeSampler(cfg, **kwargs)
    elif sampler_name in ['best', 'best_random']:
        return SimpleClickSampler(cfg, **kwargs)
    else:
        assert False, "not implemented"

	from .sampler import ShapeSampler
	from .simpleclick_sampler import SimpleClickSampler


	def build_shape_sampler(cfg, **kwargs):
	sampler_name = cfg['STROKE_SAMPLER']['EVAL']['MODE']
	if sampler_name == 'random':
	return ShapeSampler(cfg, **kwargs)
	elif sampler_name in ['best', 'best_random']:
	return SimpleClickSampler(cfg, **kwargs)
	else:
	assert False, "not implemented"