opentryon 0.0.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. opentryon-0.0.1/MANIFEST.in +7 -0
  2. opentryon-0.0.1/PKG-INFO +1470 -0
  3. opentryon-0.0.1/README.md +1402 -0
  4. opentryon-0.0.1/env.template +92 -0
  5. opentryon-0.0.1/opentryon.egg-info/PKG-INFO +1470 -0
  6. opentryon-0.0.1/opentryon.egg-info/SOURCES.txt +61 -0
  7. opentryon-0.0.1/opentryon.egg-info/dependency_links.txt +1 -0
  8. opentryon-0.0.1/opentryon.egg-info/requires.txt +29 -0
  9. opentryon-0.0.1/opentryon.egg-info/top_level.txt +2 -0
  10. opentryon-0.0.1/setup.cfg +4 -0
  11. opentryon-0.0.1/setup.py +78 -0
  12. opentryon-0.0.1/tryon/__init__.py +0 -0
  13. opentryon-0.0.1/tryon/api/__init__.py +19 -0
  14. opentryon-0.0.1/tryon/api/flux2.py +1203 -0
  15. opentryon-0.0.1/tryon/api/kling_ai.py +733 -0
  16. opentryon-0.0.1/tryon/api/lumaAI/__init__.py +22 -0
  17. opentryon-0.0.1/tryon/api/lumaAI/adapter.py +661 -0
  18. opentryon-0.0.1/tryon/api/lumaAI/luma_video_adapter.py +330 -0
  19. opentryon-0.0.1/tryon/api/nano_banana/__init__.py +17 -0
  20. opentryon-0.0.1/tryon/api/nano_banana/adapter.py +872 -0
  21. opentryon-0.0.1/tryon/api/nova_canvas.py +340 -0
  22. opentryon-0.0.1/tryon/api/segmind.py +456 -0
  23. opentryon-0.0.1/tryon/datasets/__init__.py +50 -0
  24. opentryon-0.0.1/tryon/datasets/base.py +132 -0
  25. opentryon-0.0.1/tryon/datasets/example_usage.py +324 -0
  26. opentryon-0.0.1/tryon/datasets/fashion_mnist.py +298 -0
  27. opentryon-0.0.1/tryon/datasets/subjects200k.py +604 -0
  28. opentryon-0.0.1/tryon/datasets/viton_hd.py +851 -0
  29. opentryon-0.0.1/tryon/models/__init__.py +0 -0
  30. opentryon-0.0.1/tryon/preprocessing/__init__.py +3 -0
  31. opentryon-0.0.1/tryon/preprocessing/captioning/__init__.py +2 -0
  32. opentryon-0.0.1/tryon/preprocessing/captioning/generate_caption.py +108 -0
  33. opentryon-0.0.1/tryon/preprocessing/extract_garment_new.py +91 -0
  34. opentryon-0.0.1/tryon/preprocessing/preprocess_garment.py +107 -0
  35. opentryon-0.0.1/tryon/preprocessing/preprocess_human.py +86 -0
  36. opentryon-0.0.1/tryon/preprocessing/sam2/__init__.py +23 -0
  37. opentryon-0.0.1/tryon/preprocessing/u2net/__init__.py +3 -0
  38. opentryon-0.0.1/tryon/preprocessing/u2net/data_loader.py +277 -0
  39. opentryon-0.0.1/tryon/preprocessing/u2net/load_u2net.py +47 -0
  40. opentryon-0.0.1/tryon/preprocessing/u2net/u2net_cloth_segm.py +550 -0
  41. opentryon-0.0.1/tryon/preprocessing/u2net/u2net_human_segm.py +520 -0
  42. opentryon-0.0.1/tryon/preprocessing/u2net/utils.py +10 -0
  43. opentryon-0.0.1/tryon/preprocessing/utils.py +91 -0
  44. opentryon-0.0.1/tryondiffusion/__init__.py +0 -0
  45. opentryon-0.0.1/tryondiffusion/diffusion.py +275 -0
  46. opentryon-0.0.1/tryondiffusion/ema.py +26 -0
  47. opentryon-0.0.1/tryondiffusion/network.py +865 -0
  48. opentryon-0.0.1/tryondiffusion/pre_processing/__init__.py +0 -0
  49. opentryon-0.0.1/tryondiffusion/pre_processing/generate_cloth_agnostic_rgb.py +33 -0
  50. opentryon-0.0.1/tryondiffusion/pre_processing/generate_segmented_garment.py +30 -0
  51. opentryon-0.0.1/tryondiffusion/pre_processing/openpose_pytorch/__init__.py +0 -0
  52. opentryon-0.0.1/tryondiffusion/pre_processing/openpose_pytorch/body_pose.py +233 -0
  53. opentryon-0.0.1/tryondiffusion/pre_processing/openpose_pytorch/model.py +220 -0
  54. opentryon-0.0.1/tryondiffusion/pre_processing/openpose_pytorch/utils.py +147 -0
  55. opentryon-0.0.1/tryondiffusion/pre_processing/save_pose_embeddings.py +87 -0
  56. opentryon-0.0.1/tryondiffusion/pre_processing/segment_cloth_u2net.py +33 -0
  57. opentryon-0.0.1/tryondiffusion/pre_processing/u2net_cloth_seg/__init__.py +67 -0
  58. opentryon-0.0.1/tryondiffusion/pre_processing/u2net_cloth_seg/unet.py +550 -0
  59. opentryon-0.0.1/tryondiffusion/pre_processing/u2net_cloth_seg/utils.py +85 -0
  60. opentryon-0.0.1/tryondiffusion/trainer.py +48 -0
  61. opentryon-0.0.1/tryondiffusion/utils/__init__.py +2 -0
  62. opentryon-0.0.1/tryondiffusion/utils/dataloader_train.py +92 -0
  63. opentryon-0.0.1/tryondiffusion/utils/utils.py +119 -0
opentryon-0.0.1/MANIFEST.in ADDED
@@ -0,0 +1,7 @@
1
+ include README.md
2
+ include LICENSE
3
+ include env.template
4
+ recursive-include tryon *.json *.yaml *.yml
5
+ prune tests
6
+ prune demo
7
+ prune external