clean requirements

app.py

@@ -194,14 +194,14 @@ def run_eschernet(eschernet_input_dict, sample_steps, sample_seed, nvs_num, nvs_
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
     os.makedirs(output_dir, exist_ok=True)
-    # save to N imgs
-    for i in range(T_out):
-        imsave(os.path.join(output_dir, f'{i}.png'), (image[i] * 255).astype(np.uint8))
+    # # save to N imgs
+    # for i in range(T_out):
+    #     imsave(os.path.join(output_dir, f'{i}.png'), (image[i] * 255).astype(np.uint8))
     # make a gif
     frames = [Image.fromarray((image[i] * 255).astype(np.uint8)) for i in range(T_out)]
-    frame_one = frames[0]
-    frame_one.save(os.path.join(output_dir, "output.gif"), format="GIF", append_images=frames,
-                   save_all=True, duration=50, loop=1)
+    # frame_one = frames[0]
+    # frame_one.save(os.path.join(output_dir, "output.gif"), format="GIF", append_images=frames,
+    #                save_all=True, duration=50, loop=1)
 
     # get a video
     video_path = os.path.join(output_dir, "output.mp4")

requirements.txt

@@ -1,307 +1,17 @@
-absl-py==1.4.0
-accelerate==0.23.0
-addict==2.4.0
-aiofiles==23.1.0
-aiohttp==3.8.5
-aiosignal==1.3.1
-albumentations==0.4.3
-altair==4.2.2
-annotated-types==0.5.0
-ansi2html==1.8.0
-antlr4-python3-runtime==4.9.3
-anyio==3.7.1
-appdirs==1.4.4
-asttokens==2.2.1
-async-timeout==4.0.2
-asyncer==0.0.2
-attrs==23.1.0
-backcall==0.2.0
-blinker==1.6.2
-braceexpand==0.1.7
-brotlipy==0.7.0
-cachetools==5.3.1
-carvekit_colab==4.1.2
-certifi==2024.6.2
-cffi==1.16.0
-charset-normalizer==3.3.2
-click==8.1.6
-cmake==3.29.3
-colorama==0.4.6
-coloredlogs==15.0.1
-comm==0.1.4
-ConfigArgParse==1.7
-contourpy==1.1.0
-cycler==0.11.0
-dataclasses-json==0.6.1
-datasets==2.4.0
-dearpygui==1.10.1
-decorator==4.4.2
-Deprecated==1.2.14
-diffusers==0.19.3
-dill==0.3.5.1
-docker-pycreds==0.4.0
-dpcpp-cpp-rt==2024.1.2
-easydict==1.10
-einops==0.3.0
-entrypoints==0.4
-envlight==0.1.0
-exceptiongroup==1.1.2
-executing==2.0.0
-fastapi==0.100.1
-fastcore==1.5.29
-fastjsonschema==2.18.0
-ffmpy==0.3.1
-filelock==3.14.0
-filetype==1.2.0
-fire==0.4.0
-Flask==2.2.5
-flatbuffers==23.5.26
-fonttools==4.41.1
-freetype-py==2.4.0
-frozenlist==1.4.0
-fsspec==2023.6.0
-ftfy==6.1.1
-future==0.18.3
-gitdb==4.0.10
-GitPython==3.1.32
-glfw==2.6.2
-google-auth==2.22.0
-google-auth-oauthlib==1.0.0
-gradio==4.19.2
-gradio_client==0.10.1
-grpcio==1.56.2
-h11==0.14.0
-h5py==3.10.0
-hjson==3.1.0
-httpcore==0.17.3
-httpx==0.24.1
-huggingface-hub==0.22.0
-humanfriendly==10.0
-idna==3.7
-imageio==2.32.0
-imageio-ffmpeg==0.4.8
-imgaug==0.2.6
-imgviz==1.7.3
-importlib-metadata==6.8.0
-importlib-resources==6.0.0
-iniconfig==2.0.0
-inquirerpy==0.3.4
-intel-cmplr-lib-rt==2024.1.2
-intel-cmplr-lic-rt==2024.1.2
-intel-opencl-rt==2024.1.2
-intel-openmp==2024.1.2
-intel-sycl-rt==2024.1.2
-ipython==8.18.0
-ipywidgets==8.1.0
-itsdangerous==2.1.2
-jedi==0.18.2
-Jinja2==3.1.4
-joblib==1.3.2
-jsonpatch==1.33
-jsonpointer==2.4
-jsonschema==4.18.4
-jsonschema-specifications==2023.7.1
-jupyter_core==5.3.1
-jupyterlab-widgets==3.0.8
-kiwisolver==1.4.4
-kornia==0.6.0
-lazy_loader==0.3
-libigl==2.4.1
-lightning-utilities==0.9.0
-linkify-it-py==2.0.2
-lit==18.1.6
-llvmlite==0.41.1
-loguru==0.7.0
-lovely-numpy==0.2.9
-lovely-tensors==0.1.15
-lpips==0.1.4
-Markdown==3.4.4
-markdown-it-py==2.2.0
-MarkupSafe==2.1.5
-marshmallow==3.20.1
-matplotlib==3.7.2
-matplotlib-inline==0.1.6
-mdit-py-plugins==0.3.3
-mdurl==0.1.2
-mesh2sdf==1.1.0
-mkl==2024.1.0
-mkl-fft==1.3.6
-mkl-service==2.4.0
-modelcards==0.1.6
-moviepy==1.0.3
-mpmath==1.2.1
-multidict==6.0.4
-multiprocess==0.70.13
-mypy-extensions==1.0.0
-nbformat==5.5.0
-nerfacc==0.5.2
-nest-asyncio==1.5.7
-networkx==3.3
-ninja==1.11.1
-numba==0.58.1
-numpy==1.24.4
-nvidia-cublas-cu11==11.10.3.66
-nvidia-cuda-cupti-cu11==11.7.101
-nvidia-cuda-nvrtc-cu11==11.7.99
-nvidia-cuda-runtime-cu11==11.7.99
-nvidia-cudnn-cu11==8.5.0.96
-nvidia-cufft-cu11==10.9.0.58
-nvidia-curand-cu11==10.2.10.91
-nvidia-cusolver-cu11==11.4.0.1
-nvidia-cusparse-cu11==11.7.4.91
-nvidia-nccl-cu11==2.14.3
-nvidia-nvtx-cu11==11.7.91
-oauthlib==3.2.2
-objprint==0.2.3
-omegaconf==2.3.0
-onnxruntime==1.16.1
-onnxruntime-gpu==1.16.1
-open-clip-torch==2.7.0
-open3d==0.16.0
-opencv-python==4.5.5.64
-opencv-python-headless==4.8.0.74
-orjson==3.9.2
-packaging==23.1
-pandas==2.0.3
-parso==0.8.3
-pathtools==0.1.2
-pexpect==4.8.0
-pfzy==0.3.4
-pickleshare==0.7.5
-Pillow==10.0.0
-piq==0.8.0
-platformdirs==3.10.0
-plotly==5.13.1
-pluggy==1.3.0
-plyfile==1.0.1
-pooch==1.7.0
-proglog==0.1.10
-prompt-toolkit==3.0.39
-protobuf==3.20.3
-psutil==5.9.5
-ptyprocess==0.7.0
-pudb==2019.2
-pure-eval==0.2.2
-py-cpuinfo==9.0.0
-pyarrow==12.0.1
-pyasn1==0.5.0
-pyasn1-modules==0.3.0
-pybind11==2.11.1
-pycparser==2.22
-pydantic==2.7.3
-pydantic_core==2.18.4
-pydeck==0.8.0
-pyDeprecate==0.3.1
-pydub==0.25.1
-pyglet==1.5.0
-pygltflib==1.16.1
-Pygments==2.15.1
-pyhocon==0.3.57
-PyMatting==1.1.10
-PyMCubes==0.1.2
-pymeshlab==2022.2.post4
-Pympler==1.0.1
-PyOpenGL==3.1.0
-pyparsing==3.0.9
-pypose==0.6.5
-pyquaternion==0.9.9
-pyransac3d==0.6.0
-pyre-extensions==0.0.29
-pyrender==0.1.45
-pytest==7.4.3
-python-dateutil==2.8.2
-python-multipart==0.0.9
-pytorch-fid==0.3.0
-pytorch-ignite==0.4.12
-pytorch-lightning==1.9.0
-pytz==2023.3
-pytz-deprecation-shim==0.1.0.post0
-PyWavelets==1.4.1
-PyYAML==6.0.1
-referencing==0.30.0
-regex==2023.6.3
-rembg==2.0.50
-requests==2.32.3
-requests-oauthlib==1.3.1
-responses==0.18.0
-retrying==1.3.4
-rich==13.4.2
-roma==1.5.0
-rpds-py==0.9.2
-rsa==4.9
-ruff==0.4.7
-safetensors==0.3.1
-scikit-image==0.20.0
-scikit-learn==1.3.0
-scipy==1.9.1
-segment-anything==1.0
-semantic-version==2.10.0
-sentencepiece==0.1.99
-sentry-sdk==1.28.1
-setproctitle==1.3.2
-shellingham==1.5.4
-six==1.16.0
-smmap==5.0.0
-sniffio==1.3.0
-spaces==0.28.3
-stack-data==0.6.2
-starlette==0.27.0
-streamlit==1.22.0
-style==1.1.0
-sympy==1.12.1
-taming-transformers==0.0.1
-taming-transformers-rom1504==0.0.6
-tbb==2021.12.0
-tenacity==8.2.2
-tensorboard==2.13.0
-tensorboard-data-server==0.7.1
-termcolor==2.3.0
-test_tube==0.7.5
-threadpoolctl==3.2.0
-tifffile==2023.7.18
-timm==0.9.10
-tokenizers==0.13.3
-toml==0.10.2
-tomli==2.0.1
-tomlkit==0.12.0
-toolz==0.12.0
+diffusers
 torch==2.0.0
-torch-fidelity==0.3.0
-torch_efficient_distloss==0.1.3
-torchaudio==2.0.0
-torchmetrics==1.2.0
-torchvision==0.15.1
-tornado==6.3.2
-tqdm==4.65.0
-traitlets==5.9.0
-transformers==4.30.2
-trimesh==3.23.5
-triton==2.0.0
-typeguard==2.13.3
-typer==0.12.3
-typing-inspect==0.9.0
-typing_extensions==4.12.1
-tzdata==2023.3
-tzlocal==4.3.1
-uc-micro-py==1.0.2
-update==0.0.1
-urllib3==1.26.18
-urwid==2.1.2
-uvicorn==0.23.1
-validators==0.20.0
-varname==0.12.0
-visdom==0.2.4
-wandb==0.15.7
-watchdog==3.0.0
-wcwidth==0.2.6
-webdataset==0.2.5
-websocket-client==1.6.4
-websockets==11.0.3
-Werkzeug==2.2.3
-widgetsnbextension==4.0.8
-wrapt==1.15.0
-xatlas==0.0.8
 xformers==0.0.19
-xxhash==3.2.0
-yarl==1.9.2
-zipp==3.16.2
+torchvision
+accelerate
+transformers
+gradio
+spaces
+rembg
+open3d
+trimesh
+einops
+kornia
+segment-anything
+roma
+imageio==2.32.0
+imageio-ffmpeg==0.4.8