Spaces:
Sleeping
Sleeping
Enhance README.md with inference instructions and Gradio UI details
Browse files- Added sections for CLI inference commands for both baseline and GNN checkpoints.
- Included instructions for launching the Gradio app and using it to upload images and select models.
- Updated the README to provide clearer guidance on inference processes.
README.md
CHANGED
|
@@ -125,6 +125,30 @@ Important: the processing pipeline is strict chair-only (`primary_object_class:
|
|
| 125 |
|
| 126 |
`python scripts/compare_baseline_vs_gnn.py --baseline-run smoke_baseline --gnn-run smoke_gnn`
|
| 127 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
## Expected Outputs
|
| 129 |
|
| 130 |
Data artifacts:
|
|
|
|
| 125 |
|
| 126 |
`python scripts/compare_baseline_vs_gnn.py --baseline-run smoke_baseline --gnn-run smoke_gnn`
|
| 127 |
|
| 128 |
+
### 3) Inference and Gradio UI
|
| 129 |
+
|
| 130 |
+
#### CLI inference
|
| 131 |
+
|
| 132 |
+
- Baseline checkpoint:
|
| 133 |
+
|
| 134 |
+
`python scripts/infer.py --checkpoint outputs/smoke_baseline/best.pt --image path/to/chair.jpg --top-k 5 --threshold 0.5`
|
| 135 |
+
|
| 136 |
+
- GNN checkpoint:
|
| 137 |
+
|
| 138 |
+
`python scripts/infer.py --checkpoint outputs/smoke_gnn/best.pt --image path/to/chair.jpg --top-k 5 --threshold 0.5`
|
| 139 |
+
|
| 140 |
+
#### Gradio app (local)
|
| 141 |
+
|
| 142 |
+
- Launch:
|
| 143 |
+
|
| 144 |
+
`python app.py`
|
| 145 |
+
|
| 146 |
+
- Then:
|
| 147 |
+
- Open the printed URL.
|
| 148 |
+
- Upload a chair image.
|
| 149 |
+
- Choose **baseline** or **gnn**.
|
| 150 |
+
- Adjust threshold/top-k as needed.
|
| 151 |
+
|
| 152 |
## Expected Outputs
|
| 153 |
|
| 154 |
Data artifacts:
|