sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch package

Subpackages

• sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dnnlib package
  • Submodules
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dnnlib.util module
    • EasyDict
    • Logger
      • Logger.close()
      • Logger.flush()
      • Logger.write()
    • ask_yes_no()
    • call_func_by_name()
    • construct_class_by_name()
    • copy_files_and_create_dirs()
    • format_time()
    • get_dtype_and_ctype()
    • get_module_dir_by_obj_name()
    • get_module_from_obj_name()
    • get_obj_by_name()
    • get_obj_from_module()
    • get_top_level_function_name()
    • is_pickleable()
    • is_top_level_function()
    • is_url()
    • list_dir_recursively_with_ignore()
    • make_cache_dir_path()
    • open_url()
    • set_cache_dir()
    • tuple_product()
  • Module contents
• sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics package
  • Submodules
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.frechet_inception_distance module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.inception_score module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.kernel_inception_distance module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.metric_main module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.metric_utils module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.perceptual_path_length module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.metrics.precision_recall module
  • Module contents
• sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils package
  • Subpackages
    • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops package
      • Submodules
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.bias_act module
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.conv2d_gradfix module
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.conv2d_resample module
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.fma module
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.grid_sample_gradfix module
      • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.ops.upfirdn2d module
      • Module contents
  • Submodules
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.custom_ops module
    • _find_compiler_bindir()
    • get_plugin()
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.misc module
    • InfiniteSampler
    • assert_shape()
    • check_ddp_consistency()
    • constant()
    • copy_params_and_buffers()
    • ddp_sync()
    • named_params_and_buffers()
    • params_and_buffers()
    • print_memory_diagnostics()
    • print_module_summary()
    • profiled_function()
    • reclaim_cuda_memory()
    • suppress_tracer_warnings
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.persistence module
    • _check_pickleable()
    • _module_to_src()
    • _reconstruct_persistent_obj()
    • _src_to_module()
    • import_hook()
    • is_persistent()
    • persistent_class()
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.torch_utils.training_stats module
    • Collector
      • Collector._get_delta()
      • Collector.as_dict()
      • Collector.mean()
      • Collector.names()
      • Collector.num()
      • Collector.std()
      • Collector.update()
    • _sync()
    • init_multiprocessing()
    • report0()
  • Module contents
• sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training package
  • Submodules
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.augment module
    • matrix()
    • rotate2d()
    • rotate2d_inv()
    • rotate3d()
    • scale2d()
    • scale2d_inv()
    • scale3d()
    • translate2d()
    • translate2d_inv()
    • translate3d()
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.canvas_loss module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.dataset module
    • Dataset
      • Dataset._get_raw_labels()
      • Dataset._load_raw_image()
      • Dataset._load_raw_labels()
      • Dataset.close()
      • Dataset.get_details()
      • Dataset.get_label()
      • Dataset.has_labels
      • Dataset.has_onehot_labels
      • Dataset.image_shape
      • Dataset.label_dim
      • Dataset.label_shape
      • Dataset.name
      • Dataset.num_channels
      • Dataset.original_resolution
      • Dataset.print_info()
      • Dataset.resolution
    • EnsureChannelDimension
    • ImageFolderDataset
      • ImageFolderDataset._file_ext()
      • ImageFolderDataset._get_raw_image_chw_shape()
      • ImageFolderDataset._get_zipfile()
      • ImageFolderDataset._load_raw_image()
      • ImageFolderDataset._load_raw_labels()
      • ImageFolderDataset._matches_regexp()
      • ImageFolderDataset._open_file()
      • ImageFolderDataset.close()
    • PilToNp
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.geom_loss module
    • DiceLoss
      • DiceLoss.forward()
    • GeometryLoss
      • GeometryLoss._foreground_data()
      • GeometryLoss._init_loss_dicts()
      • GeometryLoss._validate_internal_dict()
      • GeometryLoss.compute()
    • IoULoss
      • IoULoss.forward()
    • _MaskLoss
      • _MaskLoss.compute()
    • init_segmentation_loss()
    • logger
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.loss module
    • Loss
      • Loss.accumulate_gradients()
    • StyleGAN2Loss
      • StyleGAN2Loss.accumulate_gradients()
      • StyleGAN2Loss.run_D()
      • StyleGAN2Loss.run_G()
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.loss_modified module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.loss_util module
    • RGB2Gray
      • RGB2Gray.forward()
    • gradientLayer
      • gradientLayer.forward()
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.networks module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.networks_modified module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.training_loop module
  • sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.training.training_loop_modified module
  • Module contents

Submodules

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.calc_metrics module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.error(msg)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.file_ext(name: str | Path) → str

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.is_image_ext(fname: str | Path) → bool

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.make_transform(transform: str | None, output_width: int | None, output_height: int | None, resize_filter: str) → Callable[[ndarray], ndarray | None]

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.maybe_min(a: int, b: int | None) → int

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_cifar10(tarball: str, *, max_images: int | None)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_dataset(source, *, max_images: int | None)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_dest(dest: str) → Tuple[str, Callable[[str, bytes | str], None], Callable[[], None]]

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_image_folder(source_dir, *, max_images: int | None)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_image_zip(source, *, max_images: int | None)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.dataset_tool.open_lmdb(lmdb_dir: str, *, max_images: int | None)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.generate module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy module

class sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy._LegacyUnpickler(file, *, fix_imports=True, encoding='ASCII', errors='strict', buffers=())

Bases: Unpickler

find_class(module, name)

Return an object from a specified module.

If necessary, the module will be imported. Subclasses may override this method (e.g. to restrict unpickling of arbitrary classes and functions).

This method is called whenever a class or a function object is needed. Both arguments passed are str objects.

class sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy._TFNetworkStub

Bases: EasyDict

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy._collect_tf_params(tf_net)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy._populate_module_params(module, *patterns)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy.convert_tf_discriminator(tf_D)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy.convert_tf_generator(tf_G)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.legacy.load_network_pkl(f, force_fp16=False)

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.projector_orig module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.style_interpolation module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.style_mixing module

Generate style mixing image matrix using pretrained network pickle.

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.style_mixing.num_range(s: str) → List[int]

Accept either a comma separated list of numbers ‘a,b,c’ or a range ‘a-c’ and return as a list of ints.

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.train module

sketchkit.stylization.brushstrokesengine.thirdparty.stylegan2_ada_pytorch.train_orig module

Module contents