oot_diffusion

KO | EN

gitlite — search

#python #hacktoberfest #react #ai #typescript #machine-learning #rust #deep-learning #linux #nodejs #vue #awesome-list #pytorch #mcp

(★ 23)

Packaged version of OOTDiffusion

oot_diffusion 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# OOTDiffusion

A packaged version of [OOTDiffusion](https://github.com/levihsu/OOTDiffusion) that works with Pip.

No need to manually download models, checkpoints, weights etc. Should work out if the box.

Needs CUDA and GPU

OOTDiffusion 的打包版本

[Try on Replicate](https://replicate.com/viktorfa/oot_diffusion)

## Instructions

`pip install git+https://github.com/viktorfa/oot_diffusion.git`

Examples for Colab. But can be used anywhere.

If you don't set hg_root, a folder called ootd_models will be created in your working dir.

Load model
```python
from oot_diffusion import OOTDiffusionModel
from PIL import Image
from pathlib import Path

def get_ootd_model():
  model = OOTDiffusionModel(
    hg_root="/content/models",
    cache_dir="/content/drive/MyDrive/hf_cache",
  )
  return model
```

Generate image
```python
def generate_image():
  model = get_ootd_model()
  generated_images, mask_image = model.generate(
      model_path="/YOUR_MODEL.jpg",
      cloth_path="/YOUR_GARMENT.jpg",
      steps=10,
      cfg=2.0,
      num_samples=2,
    )

return generated_images, mask_image

generated_images, mask_image = generate_image()
```

Display images

```python
from IPython.display import display

for image in generated_images:
  display(image)

display(mask_image)
```

## Credits

The original authors of [OOTDiffusion](https://github.com/levihsu/OOTDiffusion)

The authors of [ComfyUI-OOTDiffusion](https://github.com/AuroBit/ComfyUI-OOTDiffusion), who made it easier to package the code.

## Official implementation

See [oms-Diffusion](https://github.com/ShineChen1024/oms-Diffusion) for the official implementation of OOTDiffusion.

oot_diffusion

oot_diffusion 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

oot_diffusion

oot_diffusion 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README