Added env specs

app.py

@@ -1,3 +1,5 @@
+"""The New Demo with html text and links and output many info and it has many customizable features"""
+
 import gdown
 import gradio as gr
 
@@ -172,7 +174,7 @@ def demo_play(img, laeo=True, rgb=False, show_keypoints=True, only_face=False, H
         #                        color=colour)
     uncertainty_mean = [i['yaw_u'] + i['pitch_u'] + i['roll_u'] for i in people_list]
     uncertainty_mean_str =  ''.join([str(round(i, 2)) + ' ' for i in uncertainty_mean])
-    return img, uncertainty_mean_str
+    return img, uncertainty_mean_str, int(person['yaw']), int(person['pitch']), int(person['roll'])
 
 
 if __name__=='__main__':
@@ -225,12 +227,17 @@ if __name__=='__main__':
 
                 with gr.Column():
                     outputs = gr.Image(label="Output Image", shape=(512, 512))
+                    with gr.Row():
+                        uncert = gr.Label(label="Uncertainty", value="0.0")
+                        yaw = gr.Label(label="Yaw", value="0")
+                        pitch = gr.Label(label="Pitch", value="0")
+                        roll = gr.Label(label="Roll", value="0")
                     uncert = gr.Label(label="Uncertainty", value="0.0")
 
         input_img.change(function_to_call, inputs=[input_img, laeo, rgb, show_keypoints, show_keypoints_only_face,
-                                                       Head_Pose_representation, detection_threshold], outputs=[outputs, uncert])
+                                                       Head_Pose_representation, detection_threshold], outputs=[outputs, uncert, yaw, pitch, roll])
         button.click(function_to_call, inputs=[input_img, laeo, rgb, show_keypoints, show_keypoints_only_face,
-                                                       Head_Pose_representation, detection_threshold], outputs=[outputs, uncert])
+                                                       Head_Pose_representation, detection_threshold], outputs=[outputs, uncert, yaw, pitch, roll])
 
         with gr.Tab("demo_upload"):
             with gr.Row():
@@ -255,7 +262,11 @@ if __name__=='__main__':
                                             label='size of displayed axis', info='size of displayed axis and cones')
                 with gr.Column():
                     outputs = gr.Image(height=238, width=585, label="Output Image")
-                    uncert = gr.Label(label="Uncertainty", value="0.0")
+                    with gr.Row():
+                        uncert = gr.Label(label="Uncertainty", value="0.0")
+                        yaw = gr.Label(label="Yaw", value="0")
+                        pitch = gr.Label(label="Pitch", value="0")
+                        roll = gr.Label(label="Roll", value="0")
                     examples_text =gr.Markdown("## Image Examples")
                     examples = gr.Examples([["LAEO_demo_data/examples/1.jpg"], ["LAEO_demo_data/examples/300wlp_0.png"],
                                 ["LAEO_demo_data/examples/AWFL_2.jpg"],
@@ -263,10 +274,10 @@ if __name__=='__main__':
 
         input_img.change(function_to_call, inputs=[input_img, laeo, rgb, show_keypoints, show_keypoints_only_face,
                                                    Head_Pose_representation, detection_threshold, thickness_points, thickness_lines, size_elements],
-                         outputs=[outputs, uncert])
+                         outputs=[outputs, uncert, yaw, pitch, roll])
         button.click(function_to_call, inputs=[input_img, laeo, rgb, show_keypoints, show_keypoints_only_face,
                                                Head_Pose_representation, detection_threshold, thickness_points, thickness_lines, size_elements],
-                     outputs=[outputs, uncert])
+                     outputs=[outputs, uncert, yaw, pitch, roll])
         # TODO create a function only to redraw last result if changed some sliders etc
 
         gr.Markdown(WEBSITE_citation)

conda_spec_file.txt

@@ -0,0 +1,80 @@
+# This file may be used to create an environment using:
+# $ conda create --name <env> --file <this file>
+# platform: linux-64
+@EXPLICIT
+https://conda.anaconda.org/conda-forge/linux-64/_libgcc_mutex-0.1-conda_forge.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/ca-certificates-2023.5.7-hbcca054_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/ld_impl_linux-64-2.40-h41732ed_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libgfortran5-12.2.0-h337968e_19.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libstdcxx-ng-12.2.0-h46fd767_19.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/python_abi-3.8-3_cp38.conda
+https://conda.anaconda.org/conda-forge/linux-64/libgfortran-ng-12.2.0-h69a702a_19.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libgomp-12.2.0-h65d4601_19.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/_openmp_mutex-4.5-2_gnu.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libgcc-ng-12.2.0-h65d4601_19.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/bzip2-1.0.8-h7f98852_4.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/cudatoolkit-11.8.0-h37601d7_11.conda
+https://conda.anaconda.org/conda-forge/linux-64/lerc-4.0.0-h27087fc_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libbrotlicommon-1.0.9-h166bdaf_8.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libdeflate-1.18-h0b41bf4_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libffi-3.4.2-h7f98852_5.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libjpeg-turbo-2.1.5.1-h0b41bf4_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libnsl-2.0.0-h7f98852_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libopenblas-0.3.21-pthreads_h78a6416_3.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libuuid-2.38.1-h0b41bf4_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libwebp-base-1.3.0-h0b41bf4_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libzlib-1.2.13-h166bdaf_4.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/ncurses-6.3-h27087fc_1.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/openssl-3.1.0-hd590300_3.conda
+https://conda.anaconda.org/conda-forge/linux-64/pthread-stubs-0.4-h36c2ea0_1001.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/xorg-libxau-1.0.11-hd590300_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/xorg-libxdmcp-1.1.3-h7f98852_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/xz-5.2.6-h166bdaf_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libblas-3.9.0-16_linux64_openblas.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libbrotlidec-1.0.9-h166bdaf_8.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libbrotlienc-1.0.9-h166bdaf_8.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libpng-1.6.39-h753d276_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libsqlite-3.42.0-h2797004_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/libxcb-1.15-h0b41bf4_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/readline-8.2-h8228510_1.conda
+https://conda.anaconda.org/conda-forge/linux-64/tk-8.6.12-h27826a3_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/zstd-1.5.2-h3eb15da_6.conda
+https://conda.anaconda.org/conda-forge/linux-64/brotli-bin-1.0.9-h166bdaf_8.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/freetype-2.12.1-hca18f0e_1.conda
+https://conda.anaconda.org/conda-forge/linux-64/libcblas-3.9.0-16_linux64_openblas.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/liblapack-3.9.0-16_linux64_openblas.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/libtiff-4.5.0-ha587672_6.conda
+https://conda.anaconda.org/conda-forge/linux-64/python-3.8.16-he550d4f_1_cpython.conda
+https://conda.anaconda.org/conda-forge/linux-64/brotli-1.0.9-h166bdaf_8.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/certifi-2023.5.7-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/charset-normalizer-3.1.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/cycler-0.11.0-pyhd8ed1ab_0.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/idna-3.4-pyhd8ed1ab_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/kiwisolver-1.4.4-py38h43d8883_1.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/lcms2-2.15-haa2dc70_1.conda
+https://conda.anaconda.org/conda-forge/noarch/munkres-1.1.4-pyh9f0ad1d_0.tar.bz2
+https://conda.anaconda.org/conda-forge/linux-64/numpy-1.24.3-py38h59b608b_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/openjpeg-2.5.0-hfec8fc6_2.conda
+https://conda.anaconda.org/conda-forge/noarch/packaging-23.1-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/pyparsing-3.0.9-pyhd8ed1ab_0.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/pysocks-1.7.1-pyha2e5f31_6.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/setuptools-67.7.2-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/six-1.16.0-pyh6c4a22f_0.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/typing_extensions-4.6.1-pyha770c72_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/unicodedata2-15.0.0-py38h0a891b7_0.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/wheel-0.40.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/zipp-3.15.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/contourpy-1.0.7-py38hfbd4bf9_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/fonttools-4.39.4-py38h01eb140_0.conda
+https://conda.anaconda.org/conda-forge/noarch/importlib_resources-5.12.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/pillow-9.5.0-py38h885162f_1.conda
+https://conda.anaconda.org/conda-forge/noarch/pip-23.1.2-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/python-dateutil-2.8.2-pyhd8ed1ab_0.tar.bz2
+https://conda.anaconda.org/conda-forge/noarch/requests-2.31.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/typing-extensions-4.6.1-hd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/importlib-resources-5.12.0-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/noarch/platformdirs-3.5.1-pyhd8ed1ab_0.conda
+https://conda.anaconda.org/conda-forge/linux-64/matplotlib-base-3.7.1-py38hd6c3c57_0.conda
+https://conda.anaconda.org/conda-forge/noarch/pooch-1.7.0-pyha770c72_3.conda
+https://conda.anaconda.org/conda-forge/linux-64/scipy-1.10.1-py38h59b608b_3.conda
+https://conda.anaconda.org/conda-forge/noarch/filterpy-1.4.5-py_1.tar.bz2

environment.yml

@@ -0,0 +1,173 @@
+name: LAEO_demo_gradio
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=conda_forge
+  - _openmp_mutex=4.5=2_gnu
+  - brotli=1.0.9=h166bdaf_8
+  - brotli-bin=1.0.9=h166bdaf_8
+  - bzip2=1.0.8=h7f98852_4
+  - ca-certificates=2023.5.7=hbcca054_0
+  - certifi=2023.5.7=pyhd8ed1ab_0
+  - charset-normalizer=3.1.0=pyhd8ed1ab_0
+  - contourpy=1.0.7=py38hfbd4bf9_0
+  - cudatoolkit=11.8.0=h37601d7_11
+  - cycler=0.11.0=pyhd8ed1ab_0
+  - filterpy=1.4.5=py_1
+  - fonttools=4.39.4=py38h01eb140_0
+  - freetype=2.12.1=hca18f0e_1
+  - idna=3.4=pyhd8ed1ab_0
+  - importlib-resources=5.12.0=pyhd8ed1ab_0
+  - importlib_resources=5.12.0=pyhd8ed1ab_0
+  - kiwisolver=1.4.4=py38h43d8883_1
+  - lcms2=2.15=haa2dc70_1
+  - ld_impl_linux-64=2.40=h41732ed_0
+  - lerc=4.0.0=h27087fc_0
+  - libblas=3.9.0=16_linux64_openblas
+  - libbrotlicommon=1.0.9=h166bdaf_8
+  - libbrotlidec=1.0.9=h166bdaf_8
+  - libbrotlienc=1.0.9=h166bdaf_8
+  - libcblas=3.9.0=16_linux64_openblas
+  - libdeflate=1.18=h0b41bf4_0
+  - libffi=3.4.2=h7f98852_5
+  - libgcc-ng=12.2.0=h65d4601_19
+  - libgfortran-ng=12.2.0=h69a702a_19
+  - libgfortran5=12.2.0=h337968e_19
+  - libgomp=12.2.0=h65d4601_19
+  - libjpeg-turbo=2.1.5.1=h0b41bf4_0
+  - liblapack=3.9.0=16_linux64_openblas
+  - libnsl=2.0.0=h7f98852_0
+  - libopenblas=0.3.21=pthreads_h78a6416_3
+  - libpng=1.6.39=h753d276_0
+  - libsqlite=3.42.0=h2797004_0
+  - libstdcxx-ng=12.2.0=h46fd767_19
+  - libtiff=4.5.0=ha587672_6
+  - libuuid=2.38.1=h0b41bf4_0
+  - libwebp-base=1.3.0=h0b41bf4_0
+  - libxcb=1.15=h0b41bf4_0
+  - libzlib=1.2.13=h166bdaf_4
+  - matplotlib-base=3.7.1=py38hd6c3c57_0
+  - munkres=1.1.4=pyh9f0ad1d_0
+  - ncurses=6.3=h27087fc_1
+  - openjpeg=2.5.0=hfec8fc6_2
+  - openssl=3.1.0=hd590300_3
+  - packaging=23.1=pyhd8ed1ab_0
+  - pillow=9.5.0=py38h885162f_1
+  - pip=23.1.2=pyhd8ed1ab_0
+  - platformdirs=3.5.1=pyhd8ed1ab_0
+  - pooch=1.7.0=pyha770c72_3
+  - pthread-stubs=0.4=h36c2ea0_1001
+  - pyparsing=3.0.9=pyhd8ed1ab_0
+  - pysocks=1.7.1=pyha2e5f31_6
+  - python=3.8.16=he550d4f_1_cpython
+  - python-dateutil=2.8.2=pyhd8ed1ab_0
+  - python_abi=3.8=3_cp38
+  - readline=8.2=h8228510_1
+  - requests=2.31.0=pyhd8ed1ab_0
+  - scipy=1.10.1=py38h59b608b_3
+  - setuptools=67.7.2=pyhd8ed1ab_0
+  - six=1.16.0=pyh6c4a22f_0
+  - tk=8.6.12=h27826a3_0
+  - typing-extensions=4.6.1=hd8ed1ab_0
+  - typing_extensions=4.6.1=pyha770c72_0
+  - unicodedata2=15.0.0=py38h0a891b7_0
+  - wheel=0.40.0=pyhd8ed1ab_0
+  - xorg-libxau=1.0.11=hd590300_0
+  - xorg-libxdmcp=1.1.3=h7f98852_0
+  - xz=5.2.6=h166bdaf_0
+  - zipp=3.15.0=pyhd8ed1ab_0
+  - zstd=1.5.2=h3eb15da_6
+  - pip:
+      - absl-py==1.4.0
+      - aiofiles==23.2.1
+      - aiohttp==3.8.5
+      - aiosignal==1.3.1
+      - altair==5.0.1
+      - annotated-types==0.5.0
+      - anyio==3.7.1
+      - astunparse==1.6.3
+      - async-timeout==4.0.3
+      - attrs==23.1.0
+      - beautifulsoup4==4.12.2
+      - cachetools==5.3.0
+      - click==8.1.7
+      - cython==0.29.34
+      - exceptiongroup==1.1.3
+      - fastapi==0.101.1
+      - ffmpy==0.3.1
+      - flatbuffers==23.5.9
+      - frozenlist==1.4.0
+      - fsspec==2023.6.0
+      - gast==0.4.0
+      - gdown==4.7.1
+      - google-auth==2.18.1
+      - google-auth-oauthlib==1.0.0
+      - google-pasta==0.2.0
+      - gradio==3.40.1
+      - gradio-client==0.4.0
+      - grpcio==1.54.2
+      - h11==0.14.0
+      - h5py==3.8.0
+      - httpcore==0.17.3
+      - httpx==0.24.1
+      - huggingface-hub==0.16.4
+      - importlib-metadata==6.6.0
+      - jax==0.4.10
+      - jinja2==3.1.2
+      - jsonschema==4.19.0
+      - jsonschema-specifications==2023.7.1
+      - keras==2.12.0
+      - libclang==16.0.0
+      - linkify-it-py==2.0.2
+      - markdown==3.4.3
+      - markdown-it-py==2.2.0
+      - markupsafe==2.1.2
+      - mdit-py-plugins==0.3.3
+      - mdurl==0.1.2
+      - ml-dtypes==0.1.0
+      - multidict==6.0.4
+      - numpy==1.23.5
+      - nvidia-cublas-cu11==11.11.3.6
+      - nvidia-cudnn-cu11==8.6.0.163
+      - oauthlib==3.2.2
+      - opencv-python==4.7.0.72
+      - opt-einsum==3.3.0
+      - orjson==3.9.5
+      - pandas==2.0.3
+      - pkgutil-resolve-name==1.3.10
+      - protobuf==4.23.1
+      - pyasn1==0.5.0
+      - pyasn1-modules==0.3.0
+      - pydantic==2.2.1
+      - pydantic-core==2.6.1
+      - pydub==0.25.1
+      - pyopengl==3.1.7
+      - python-multipart==0.0.6
+      - pytz==2023.3
+      - pyyaml==6.0.1
+      - referencing==0.30.2
+      - requests-oauthlib==1.3.1
+      - rpds-py==0.9.2
+      - rsa==4.9
+      - semantic-version==2.10.0
+      - sniffio==1.3.0
+      - soupsieve==2.4.1
+      - starlette==0.27.0
+      - tensorboard==2.12.3
+      - tensorboard-data-server==0.7.0
+      - tensorflow==2.12.0
+      - tensorflow-estimator==2.12.0
+      - tensorflow-io-gcs-filesystem==0.32.0
+      - termcolor==2.3.0
+      - toolz==0.12.0
+      - tqdm==4.66.1
+      - tzdata==2023.3
+      - uc-micro-py==1.0.2
+      - urllib3==1.26.16
+      - uvicorn==0.23.2
+      - websockets==11.0.3
+      - werkzeug==2.3.4
+      - wrapt==1.14.1
+      - yarl==1.9.2
+prefix: /home/federico/miniconda3/envs/LAEO_demo_gradio