MCPcopy Create free account

hub / github.com/bcmi/OSInsert-Image-Composition / functions

Functions545 in github.com/bcmi/OSInsert-Image-Composition

⨍ Functions 545◇ Types & classes 112
Methodon_train_batch_end
(self, *args, **kwargs)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:394
Methodon_train_batch_start
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:506
Methodon_train_epoch_start
(self)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:157
Methodon_validation_start
(self)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:206
Functionparallel_data_prefetch
( func: callable, data, n_proc, target_data_type="ndarray", cpu_intensive=True, use_worker_id=False )
libcom/os_insert/source/ldm/util.py:108
Functionpick_and_pop
(keys, d)
libcom/os_insert/source/ldm/modules/x_transformer.py:88
Functionpoisson_blending
(fg, fg_mask, bg, center=None)
libcom/os_insert/source/ldm/data/open_images.py:339
Methodprepare_data
(self)
libcom/os_insert/source/ldm/data/datamodule.py:75
Methodprogressive_denoising
(self, cond, shape, verbose=True, callback=None, quantize_denoised=False, img_ca
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1037
Functionrandom_crop_image
(image, crop_w, crop_h)
libcom/os_insert/source/ldm/data/open_images.py:317
Functionread_image_bgr
(path: str | Path)
libcom/os_insert/source/utils.py:36
Methodreformat_tensor
(x)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:336
Methodregister_schedule
(self, given_betas=None, beta_schedule="linear", timesteps=1000,
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:521
Methodrescale_bbox
(bbox)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:896
Functionrun_objectstitch_single
Run ObjectStitch for a single sample and return the coarse composite path. This wraps the original ObjectStitch `scripts/inference.py` logic into
libcom/os_insert/source/objectstitch_infer.py:328
Functionrun_objectstitch_single_image
Run ObjectStitch for a single sample and return the coarse composite image (RGB uint8).
libcom/os_insert/source/objectstitch_infer.py:411
Functionrun_objectstitch_single_image_from_images
Run ObjectStitch for a single sample and return the coarse composite image (RGB uint8). This is the in-memory variant of :func:`run_objectstitch_
libcom/os_insert/source/objectstitch_infer.py:563
Functionrun_sam_on_objectstitch
( os_image: np.ndarray, bg_shape_hw: Tuple[int, int], bbox_xyxy_bg: Tuple[int, int, int, int],
libcom/os_insert/source/sam_on_objectstitch.py:148
Functionrun_sam_on_objectstitch_single
Run SAM on a single ObjectStitch composite and return a binary mask path. Parameters ---------- os_image_path: Path to the Object
libcom/os_insert/source/sam_on_objectstitch.py:69
Methodsample
(self)
libcom/os_insert/source/ldm/modules/distributions/distributions.py:6
Methodsample
(self)
libcom/os_insert/source/ldm/modules/distributions/distributions.py:17
Methodsample
(self)
libcom/os_insert/source/ldm/modules/distributions/distributions.py:35
Methodsample
(self, S, batch_size, shape, conditioning=None,
libcom/os_insert/source/ldm/models/diffusion/plms.py:58
Functionscale_module
Scale the parameters of a module and return it.
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:183
Functionscan_all_files
()
libcom/os_insert/source/ldm/data/open_images.py:76
Methodschedule
(self, n, **kwargs)
libcom/os_insert/source/ldm/lr_scheduler.py:83
Functionsd2clip
(x)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1334
Methodsetup
(self, stage=None)
libcom/os_insert/source/ldm/data/datamodule.py:79
Methodshared_step
(self, batch, **kwargs)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:876
Methodstochastic_encode
(self, x0, t, use_original_steps=False, noise=None)
libcom/os_insert/source/ldm/models/diffusion/ddim.py:229
Functionstring_begins_with
(prefix, str)
libcom/os_insert/source/ldm/modules/x_transformer.py:102
Functiontest_cocoee_dataset
()
libcom/os_insert/source/ldm/data/open_images.py:619
Functiontest_open_images
()
libcom/os_insert/source/ldm/data/open_images.py:654
Functiontest_open_images_efficiency
()
libcom/os_insert/source/ldm/data/open_images.py:694
Methodtraining_step
(self, batch, batch_idx, optimizer_idx)
libcom/os_insert/source/ldm/models/autoencoder.py:142
Methodtraining_step
(self, batch, batch_idx, optimizer_idx)
libcom/os_insert/source/ldm/models/autoencoder.py:351
Methodtraining_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:198
Methodtraining_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:370
Functionuniform_on_device
(r1, r2, shape, device)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:45
Functionuniq
(arr)
libcom/os_insert/source/ldm/modules/attention.py:15
Methodvalidation_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/autoencoder.py:164
Methodvalidation_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/autoencoder.py:372
Methodvalidation_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:210
Methodvalidation_step
(self, batch, batch_idx)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:386
Functionworker_init_fn
(_)
libcom/os_insert/source/ldm/data/datamodule.py:37
← previous501–545 of 545, ranked by callers