A User Interface for DETR built with Dash. 100% Python.
A User Interface for DETR built with Dash. 100% Python. Click here for a demo.
The release of DETR: End-to-End Object Detection with Transformers showed significant improvement in real-time object detection and panoptic segmentation (PS), while greatly simplifying the architecture. As a mean to test the model, we decided to build a simple Dash app that let you experiment and play with the model through a user interface.
Clone this repo:
git clone https://github.com/plotly/dash-detr cd dash-detr
Create a fresh venv (with
condaor
virtualenv) and activate it:
conda create -n dash-detr python=3.7 conda activate dash-detr
Install the requirements:
pip install -r requirements.txt
Start the app:
python app.py
Try the app at
localhost:8050!
To make it easy for you to extend the app by adding a custom model, we placed everything related to PyTorch and modeling in
model.py, so you can add your own models without modifying the app. If you want to customize the layout or create new interactions, you can edit
app.py(if you are not familiar with Dash, read the tutorials first.)
If you are interested in deploy apps like this one for production, check out our article on productionizing object detection models with Dash Enterprise, or reach out to us.