def test_demo_images_guides_smoke(name, regions): image = demo.demo_images()[name] assert isinstance(image.guides, data.DownloadableImageCollection) names, _ = zip(*iter(image.guides)) assert set(names).issuperset(set(regions))
def test_demo_images_smoke(self): with get_tmp_dir() as root: images = demo.demo_images() images.download(root=root) images = [image for _, image in images] guides = [ [guide for _, guide in image.guides] for image in images if image.guides is not None ] for image_or_guide in itertools.chain(images, *guides): with self.subTest(image_or_guide=image_or_guide): file = path.join(root, image_or_guide.file) self.assertTrue( path.exists(file), msg=f"File {file} does not exist." ) actual = calculate_md5(file) desired = image_or_guide.md5 self.assertEqual( actual, desired, msg=( f"The actual and desired MD5 hash of the image mismatch: " f"{actual} != {desired}" ), )
def test_demo_images_smoke(): images = demo.demo_images() assert isinstance(images, data.DownloadableImageCollection) names, _ = zip(*iter(images)) assert set(names) == { "bird1", "paint", "bird2", "mosaic", "castle", "church", "cliff", }
from pystiche.loss import PerceptualLoss from pystiche.misc import get_device, get_input_image from pystiche.ops import ( FeatureReconstructionOperator, MRFOperator, MultiLayerEncodingOperator, ) from pystiche.optim import default_image_pyramid_optim_loop from pystiche.pyramid import ImagePyramid print(f"I'm working with pystiche=={pystiche.__version__}") device = get_device() print(f"I'm working with {device}") images = demo_images() images.download() ######################################################################################## # At first we define a :class:`~pystiche.loss.perceptual.PerceptualLoss` that is used # as optimization ``criterion``. multi_layer_encoder = vgg19_multi_layer_encoder() content_layer = "relu4_2" content_encoder = multi_layer_encoder.extract_single_layer_encoder( content_layer) content_weight = 1e0 content_loss = FeatureReconstructionOperator(content_encoder, score_weight=content_weight)
from os import path import pytest from pystiche.demo import demo_images from .utils import ( assert_downloads_correctly, assert_is_downloadable, rate_limited_urlopen, retry, ) from tests.mocks import make_mock_target IMAGES_AND_IDS = [(image, f"demo, {name}") for name, image in itertools.chain(demo_images(), )] IMAGE_PARAMETRIZE_KWARGS = dict(zip(("argvalues", "ids"), zip(*IMAGES_AND_IDS))) def assert_image_is_downloadable(image, **kwargs): assert_is_downloadable(image.url, **kwargs) @pytest.mark.slow @pytest.mark.parametrize("image", **IMAGE_PARAMETRIZE_KWARGS) def test_image_download_smoke(subtests, image): retry(lambda: assert_image_is_downloadable(image), times=2, wait=5.0)