Update README.md
Browse files
README.md
CHANGED
|
@@ -133,7 +133,7 @@ The training process demonstrates excellent convergence with:
|
|
| 133 |
```python
|
| 134 |
pip install -q rfdetr==1.2.1 supervision==0.26.1
|
| 135 |
```
|
| 136 |
-
- You can use: [video_processing.py](https://huggingface.co/Subh775/Threat-Detection-
|
| 137 |
|
| 138 |
- Below is the script to process a single image
|
| 139 |
|
|
@@ -157,7 +157,7 @@ THREAT_CLASSES = {
|
|
| 157 |
image = Image.open("Path_to_image")
|
| 158 |
|
| 159 |
# pre-trained weights
|
| 160 |
-
weights_url = "https://huggingface.co/Subh775/Threat-Detection-
|
| 161 |
weights_filename = "checkpoint_best_total.pth"
|
| 162 |
|
| 163 |
# Download weights if not already present
|
|
|
|
| 133 |
```python
|
| 134 |
pip install -q rfdetr==1.2.1 supervision==0.26.1
|
| 135 |
```
|
| 136 |
+
- You can use: [video_processing.py](https://huggingface.co/Subh775/Threat-Detection-RFDETR/blob/main/video_processing.py) to process large videos
|
| 137 |
|
| 138 |
- Below is the script to process a single image
|
| 139 |
|
|
|
|
| 157 |
image = Image.open("Path_to_image")
|
| 158 |
|
| 159 |
# pre-trained weights
|
| 160 |
+
weights_url = "https://huggingface.co/Subh775/Threat-Detection-RFDETR/resolve/main/checkpoint_best_total.pth"
|
| 161 |
weights_filename = "checkpoint_best_total.pth"
|
| 162 |
|
| 163 |
# Download weights if not already present
|