MCPcopy Create free account

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

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

↓ 2 callersFunctionget_bbox_from_mask
Get (y1, y2, x1, x2) bounding box from a binary mask. This mirrors the original InsertAnything implementation: when the mask is nearly empty,
libcom/os_insert/source/ia_utils.py:16
↓ 2 callersFunctionget_bbox_tensor
Normalize bbox coordinates to [0,1] as in ObjectStitch.
libcom/os_insert/source/objectstitch_data.py:60
↓ 2 callersMethodget_conditioning
(self, batch, k=None)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:133
↓ 2 callersMethodget_input
(self, batch, k)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:124
↓ 2 callersMethodget_learned_conditioning
(self, c)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:582
↓ 2 callersFunctionget_tensor
Basic image -> tensor transform used by ObjectStitch (Stable Diffusion style). This is a minimal copy of the preprocessing from ObjectStitch's
libcom/os_insert/source/objectstitch_data.py:11
↓ 2 callersFunctiongroup_dict_by_key
(cond, d)
libcom/os_insert/source/ldm/modules/x_transformer.py:93
↓ 2 callersFunctiongroupby_prefix_and_trim
(prefix, d)
libcom/os_insert/source/ldm/modules/x_transformer.py:110
↓ 2 callersMethodinfer_images
( self, *, background: np.ndarray, foreground: np.ndarray, foreground_
libcom/os_insert/os_insert.py:315
↓ 2 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:192
↓ 2 callersFunctionload_bbox_txt
Load a simple `x1 y1 x2 y2` bbox txt file.
libcom/os_insert/source/utils.py:12
↓ 2 callersFunctionmake_ddim_sampling_parameters
(alphacums, ddim_timesteps, eta, verbose=True)
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:63
↓ 2 callersFunctionmake_ddim_timesteps
(ddim_discr_method, num_ddim_timesteps, num_ddpm_timesteps, verbose=True)
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:46
↓ 2 callersMethodp_losses
(self, inputs, t, noise=None, )
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:932
↓ 2 callersMethodp_sample
(self, x, bbox, c, t, clip_denoised=False, repeat_noise=False, return_codebook_ids=False, qua
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1006
↓ 2 callersMethodp_sample_ddim
(self, x, c, t, index, mask=None, repeat_noise=False, use_original_steps=False, quantize_denoised=False,
libcom/os_insert/source/ldm/models/diffusion/ddim.py:176
↓ 2 callersMethodpredict_start_from_noise
(self, x_t, t, noise)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:222
↓ 2 callersMethodq_posterior
(self, x_start, x_t, t)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:228
↓ 2 callersMethodregister_schedule
(self, given_betas=None, beta_schedule="linear", timesteps=1000, linear_start=1e-4,
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:123
↓ 2 callersMethodrestore
Restore the parameters stored with the `store` method. Useful to validate the model with EMA parameters without affecting the
libcom/os_insert/source/ldm/modules/ema.py:64
↓ 2 callersFunctionreverse_image_tensor
(tensor, img_size=(256,256))
libcom/os_insert/source/ldm/data/open_images.py:269
↓ 2 callersFunctionreverse_mask_tensor
(tensor, img_size=(256,256))
libcom/os_insert/source/ldm/data/open_images.py:284
↓ 2 callersFunctionrun_objectstitch_single_image_from_images_cached
Cached in-memory ObjectStitch inference. This avoids re-loading weights for every call.
libcom/os_insert/source/objectstitch_infer.py:512
↓ 2 callersFunctionrun_sam_on_objectstitch_with_predictor
( *, predictor, os_image: np.ndarray, bg_shape_hw: Tuple[int, int], bbox_xyxy_bg: Tuple[in
libcom/os_insert/source/sam_on_objectstitch.py:218
↓ 2 callersMethodstore
Save the current parameters for restoring later. Args: parameters: Iterable of `torch.nn.Parameter`; the parameters to be
libcom/os_insert/source/ldm/modules/ema.py:55
↓ 2 callersFunctiontimestep_embedding
Create sinusoidal timestep embeddings. :param timesteps: a 1-D Tensor of N indices, one per batch element. These may be
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:151
↓ 2 callersMethodto_rgb
(self, x)
libcom/os_insert/source/ldm/models/autoencoder.py:417
↓ 2 callersMethodto_rgb
(self, x)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1287
↓ 2 callersFunctionvis_random_augtype
(batch)
libcom/os_insert/source/ldm/data/open_images.py:592
↓ 1 callersMethod__init__
(self, dataset)
libcom/os_insert/source/ldm/data/datamodule.py:28
↓ 1 callersMethod__len__
(self)
libcom/os_insert/source/ldm/data/base.py:18
↓ 1 callersMethod_encode_vae_image
(self, image: torch.Tensor, generator: torch.Generator)
diffusers_osinsert/_patched_pipeline_flux_fill.py:423
↓ 1 callersFunction_ensure_mask_uint8
(mask: np.ndarray | Image.Image)
libcom/os_insert/source/objectstitch_infer.py:211
↓ 1 callersFunction_f
(r, T=0.6, beta=0.1)
libcom/os_insert/source/ia_utils.py:48
↓ 1 callersMethod_filter_relpaths
(self, relpaths)
libcom/os_insert/source/ldm/data/imagenet.py:48
↓ 1 callersMethod_get_clip_prompt_embeds
( self, prompt: Union[str, List[str]], num_images_per_prompt: int = 1, device:
diffusers_osinsert/_patched_pipeline_flux_fill.py:263
↓ 1 callersMethod_get_t5_prompt_embeds
( self, prompt: Union[str, List[str]] = None, num_images_per_prompt: int = 1,
diffusers_osinsert/_patched_pipeline_flux_fill.py:215
↓ 1 callersMethod_load
(self)
libcom/os_insert/source/ldm/data/imagenet.py:93
↓ 1 callersFunction_load_patched_call
()
diffusers_osinsert/__init__.py:20
↓ 1 callersFunction_load_patched_pipeline_source
()
diffusers_osinsert/__init__.py:10
↓ 1 callersFunction_normalize_suffix
(s: str)
libcom/os_insert/source/insertanything_infer.py:169
↓ 1 callersFunction_parse_csv_floats
(s: str | None)
tests/test_os_insert.py:148
↓ 1 callersMethod_prepare
(self)
libcom/os_insert/source/ldm/data/imagenet.py:45
↓ 1 callersMethod_prepare_human_to_integer_label
(self)
libcom/os_insert/source/ldm/data/imagenet.py:80
↓ 1 callersMethod_prepare_idx_to_synset
(self)
libcom/os_insert/source/ldm/data/imagenet.py:74
↓ 1 callersMethod_prepare_latent_image_ids
(batch_size, height, width, device, dtype)
diffusers_osinsert/_patched_pipeline_flux_fill.py:512
↓ 1 callersMethod_prepare_synset_to_human
(self)
libcom/os_insert/source/ldm/data/imagenet.py:66
↓ 1 callersFunction_resolve_ia_paths
( model_dir: str | Path | None, *, flux_fill_path: str | Path | None = None, flux_redux_path:
libcom/os_insert/source/insertanything_infer.py:36
↓ 1 callersMethod_unpack_latents
(latents, height, width, vae_scale_factor)
diffusers_osinsert/_patched_pipeline_flux_fill.py:532
↓ 1 callersMethodaugment_foreground
(self, img, mask)
libcom/os_insert/source/ldm/data/open_images.py:158
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:238
↓ 1 callersFunctionbbox2mask
(bbox, mask_w, mask_h)
libcom/os_insert/source/ldm/data/open_images.py:189
↓ 1 callersFunctionbox2squre
Convert a bbox to square, matching the original InsertAnything logic.
libcom/os_insert/source/ia_utils.py:104
↓ 1 callersFunctionbuild_sam_predictor
(*, config: SamOnObjectStitchConfig)
libcom/os_insert/source/sam_on_objectstitch.py:198
↓ 1 callersMethodcalculate_adaptive_weight
(self, nll_loss, g_loss, last_layer=None)
libcom/os_insert/source/ldm/modules/losses/vqperceptual.py:85
↓ 1 callersMethodcalculate_adaptive_weight
(self, nll_loss, g_loss, last_layer=None)
libcom/os_insert/source/ldm/modules/losses/contperceptual.py:32
↓ 1 callersFunctioncalculate_shift
( image_seq_len, base_seq_len: int = 256, max_seq_len: int = 4096, base_shift: float = 0.5,
diffusers_osinsert/_patched_pipeline_flux_fill.py:80
↓ 1 callersMethodcheck_inputs
( self, prompt, prompt_2, strength, height, width, pro
diffusers_osinsert/_patched_pipeline_flux_fill.py:446
↓ 1 callersFunctionclip2sd
(x)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1325
↓ 1 callersFunctionconstant_pad_bbox
(bbox, width, height, value=10)
libcom/os_insert/source/ldm/data/open_images.py:206
↓ 1 callersFunctioncrop_back
Paste the edited crop back into the original target image. This is a faithful port of the original InsertAnything geometry, including margin
libcom/os_insert/source/ia_utils.py:127
↓ 1 callersFunctioncrop_foreground_by_bbox
(img, mask, bbox, pad_bbox=10)
libcom/os_insert/source/ldm/data/open_images.py:237
↓ 1 callersMethodddim_sampling
(self, cond, shape, x_T=None, ddim_use_original_steps=False, callb
libcom/os_insert/source/ldm/models/diffusion/ddim.py:115
↓ 1 callersFunctiondisable_patch
Disable the OSInsert FluxFillPipeline patch and restore upstream behavior.
diffusers_osinsert/__init__.py:53
↓ 1 callersFunctionenable_patch
Enable the OSInsert FluxFillPipeline patch for the current Python process.
diffusers_osinsert/__init__.py:39
↓ 1 callersMethodencode
(self, text)
libcom/os_insert/source/ldm/modules/encoders/modules.py:102
↓ 1 callersMethodencode
(self, x)
libcom/os_insert/source/ldm/models/autoencoder.py:324
↓ 1 callersMethodencode_prompt
( self, prompt: Union[str, List[str]], prompt_2: Union[str, List[str]], device
diffusers_osinsert/_patched_pipeline_flux_fill.py:371
↓ 1 callersMethodencode_with_pretrained
(self,x)
libcom/os_insert/source/ldm/modules/diffusionmodules/model.py:816
↓ 1 callersFunctionequals
(val)
libcom/os_insert/source/ldm/modules/x_transformer.py:76
↓ 1 callersFunctionexpand_image_mask
Expand image and mask by symmetric padding (original behavior).
libcom/os_insert/source/ia_utils.py:169
↓ 1 callersMethodforward
(self, x)
libcom/os_insert/source/ldm/modules/diffusionmodules/util.py:210
↓ 1 callersMethodforward
(self, x)
libcom/os_insert/source/ldm/modules/encoders/xf.py:56
↓ 1 callersMethodfreeze
(self)
libcom/os_insert/source/ldm/modules/encoders/modules.py:154
↓ 1 callersMethodget_base
(self)
libcom/os_insert/source/ldm/data/imagenet.py:379
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
libcom/os_insert/source/objectstitch_infer.py:85
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
libcom/os_insert/source/ldm/util.py:88
↓ 1 callersFunctionget_tensor_clip
Foreground (CLIP-style) image transform from ObjectStitch.
libcom/os_insert/source/objectstitch_data.py:33
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
libcom/os_insert/source/ldm/modules/diffusionmodules/model.py:12
↓ 1 callersMethodget_timesteps
(self, num_inference_steps, strength, device)
diffusers_osinsert/_patched_pipeline_flux_fill.py:436
↓ 1 callersMethodget_x_noisy
(self, x, t, noise=None)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:110
↓ 1 callersMethodinit_
(self)
libcom/os_insert/source/ldm/modules/x_transformer.py:31
↓ 1 callersMethodinit_
(self)
libcom/os_insert/source/ldm/modules/x_transformer.py:595
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
libcom/os_insert/source/ldm/models/autoencoder.py:78
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
libcom/os_insert/source/ldm/models/autoencoder.py:313
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:70
↓ 1 callersMethodinstantiate_cond_stage
(self, config)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:537
↓ 1 callersMethodinstantiate_first_stage
(self, config)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:530
↓ 1 callersMethodinstantiate_pretrained
(self, config)
libcom/os_insert/source/ldm/modules/diffusionmodules/model.py:807
↓ 1 callersFunctionismap
(x)
libcom/os_insert/source/ldm/util.py:41
↓ 1 callersMethodkl
(self, other=None)
libcom/os_insert/source/ldm/modules/distributions/distributions.py:39
↓ 1 callersMethodload_bbox_file
(self, bbox_file)
libcom/os_insert/source/ldm/data/open_images.py:380
↓ 1 callersMethodload_bbox_file
(self, bbox_file)
libcom/os_insert/source/ldm/data/open_images.py:541
↓ 1 callersMethodload_bbox_path_list
(self)
libcom/os_insert/source/ldm/data/open_images.py:364
↓ 1 callersMethodload_classifier
(self, ckpt_path, pool)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:95
↓ 1 callersMethodload_diffusion
(self)
libcom/os_insert/source/ldm/models/diffusion/classifier.py:88
↓ 1 callersFunctionload_objectstitch_model_and_sampler
Load ObjectStitch model + sampler once. The returned model is moved to ``config.device`` and set to ``eval()``.
libcom/os_insert/source/objectstitch_infer.py:480
↓ 1 callersMethodlog_images
(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs)
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:406
↓ 1 callersFunctionlog_local
(save_dir, split, images, global_step, current_epo
libcom/os_insert/source/ldm/models/diffusion/ddpm.py:1351
↓ 1 callersFunctionlog_txt_as_img
(wh, xc, size=10)
libcom/os_insert/source/ldm/util.py:17
← previousnext →101–200 of 545, ranked by callers