Add application file

app.py

@@ -0,0 +1,44 @@
+import gradio as gr
+from deepforest import main
+import cv2
+
+
+def show_trees(img_path):
+    model = main.deepforest()
+    model.use_release()
+    img = model.predict_image(path=img_path, return_plot=True)
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return img
+
+def show_birds(img_path):
+    model = main.deepforest()
+    model.use_bird_release()
+    img = model.predict_image(path=img_path, return_plot=True)
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return img
+
+with gr.Blocks() as demo:
+    gr.Markdown('# Deepforest')
+    gr.Markdown('## Tree Detection Model')
+    gr.Markdown('### Predict trees')
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="Input Image", type="filepath")
+        with gr.Column():
+            output_image = gr.Image(label="Predicted image")
+
+    submit_button_trees = gr.Button("Predict trees")
+    submit_button_trees.click(show_trees, inputs=input_image, outputs=output_image)
+    
+    gr.Markdown('### Predict birds')
+    with gr.Row():
+        with gr.Column():
+            input_image=gr.Image(label="Input image",type="filepath")
+        with gr.Column():
+            output_image=gr.Image(label="Predicted Image")
+            
+    submit_button_birds = gr.Button("Predict birds")
+    submit_button_birds.click(show_birds,inputs=input_image,outputs=output_image)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,136 @@
+affine @ file:///Users/ktietz/demo/mc3/conda-bld/affine_1629890479180/work
+aiofiles==23.2.1
+aiohttp==3.9.3
+aiosignal==1.3.1
+alabaster==0.7.16
+albumentations==1.4.2
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.3.0
+attrs @ file:///croot/attrs_1695717823297/work
+Babel==2.14.0
+certifi @ file:///croot/certifi_1707229174982/work/certifi
+charset-normalizer==3.3.2
+click @ file:///croot/click_1698129812380/work
+click-plugins @ file:///Users/ktietz/demo/mc3/conda-bld/click-plugins_1630665883852/work
+cligj==0.7.2
+colorama==0.4.6
+commonmark==0.9.1
+contourpy==1.2.0
+cycler==0.12.1
+docutils==0.20.1
+fastapi==0.110.0
+ffmpy==0.3.2
+filelock==3.13.3
+fiona==1.9.6
+fonttools==4.50.0
+frozenlist==1.4.1
+fsspec==2024.3.1
+geopandas==0.14.3
+gradio==4.23.0
+gradio_client==0.14.0
+h11==0.14.0
+httpcore==1.0.4
+httpx==0.27.0
+huggingface-hub==0.22.1
+idna==3.6
+imagecodecs==2024.1.1
+imageio==2.34.0
+imagesize==1.4.1
+importlib_resources==6.4.0
+Jinja2==3.1.3
+joblib==1.3.2
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+lazy_loader==0.3
+lightning-utilities==0.11.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.8.3
+mdurl==0.1.2
+mkl-fft @ file:///croot/mkl_fft_1695058164594/work
+mkl-random @ file:///croot/mkl_random_1695059800811/work
+mkl-service==2.4.0
+mpmath==1.3.0
+multidict==6.0.5
+networkx==3.2.1
+numpy @ file:///croot/numpy_and_numpy_base_1708638617955/work/dist/numpy-1.26.4-cp311-cp311-linux_x86_64.whl#sha256=5f96f274d410a1682519282ae769c877d32fdbf171aa8badec7bf5e1d3a1748a
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.19.3
+nvidia-nvjitlink-cu12==12.4.99
+nvidia-nvtx-cu12==12.1.105
+opencv-python==4.9.0.80
+opencv-python-headless==4.9.0.80
+orjson==3.9.15
+packaging==24.0
+pandas==2.2.1
+pillow==10.2.0
+progressbar2==4.4.2
+psutil==5.9.8
+pycocotools==2.0.7
+pydantic==2.6.4
+pydantic_core==2.16.3
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing @ file:///work/ci_py311/pyparsing_1677811559502/work
+pyproj==3.6.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.9
+python-utils==3.8.2
+pytorch-lightning==2.2.1
+pytz==2024.1
+PyYAML==6.0.1
+rasterio @ file:///work/ci_py311/rasterio_1676861695610/work
+recommonmark==0.7.1
+referencing==0.34.0
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.0
+Rtree==1.2.0
+ruff==0.3.4
+scikit-image==0.22.0
+scikit-learn==1.4.1.post1
+scipy==1.12.0
+semantic-version==2.10.0
+shapely==2.0.3
+shellingham==1.5.4
+six==1.16.0
+slidingwindow==0.0.14
+sniffio==1.3.1
+snowballstemmer==2.2.0
+snuggs @ file:///Users/ktietz/demo/mc3/conda-bld/snuggs_1629803886966/work
+Sphinx==7.2.6
+sphinxcontrib-applehelp==1.0.8
+sphinxcontrib-devhelp==1.0.6
+sphinxcontrib-htmlhelp==2.0.5
+sphinxcontrib-jsmath==1.0.1
+sphinxcontrib-qthelp==1.0.7
+sphinxcontrib-serializinghtml==1.1.10
+starlette==0.36.3
+sympy==1.12
+threadpoolctl==3.4.0
+tifffile==2024.2.12
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.2.1
+torchmetrics==1.3.2
+torchvision==0.17.1
+tqdm==4.66.2
+triton==2.2.0
+typer==0.10.0
+typing_extensions==4.10.0
+tzdata==2024.1
+urllib3==2.2.1
+uvicorn==0.29.0
+websockets==11.0.3
+xmltodict==0.13.0
+yarl==1.9.4