Commit 2493089
authored
feat: super gradients integration (#246)
Add new detection class which allows the user to supply a
super-gradients model with the following values;
1. model_arch supported by super-gradients
2. model_path to the model weights
3. yaml file with a names section containing in order the labels for the
classes to the object detection model
4. callback function which takes an image and a super-gradients model
and returns supervision detections
Wlll be used for AF contract and contracts in the future.1 parent ae5c2a8 commit 2493089
File tree
12 files changed
+443
-58
lines changed- requirements
- test_unstructured_inference/models
- unstructured_inference
- inference
- models
12 files changed
+443
-58
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
1 | 5 | | |
2 | 6 | | |
3 | 7 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
5 | | - | |
| 5 | + | |
| 6 | + | |
6 | 7 | | |
7 | 8 | | |
8 | 9 | | |
| |||
0 commit comments