Skip to content

Commit 2493089

Browse files
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

12 files changed

+443
-58
lines changed

CHANGELOG.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
1+
## 0.7.4-dev0
2+
3+
* Allows user to use super-gradients model if they have a callback predict function, a yaml file with names field corresponding to classes and a path to the model weights
4+
15
## 0.7.3
26

37
* Integration of Chipperv2 and additional Chipper functionality, which includes automatic detection of GPU,

requirements/base.in

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@ layoutparser[layoutmodels,tesseract]
22
python-multipart
33
huggingface-hub
44
opencv-python!=4.7.0.68
5-
onnx==1.14.1
5+
super-gradients
6+
supervision
67
# NOTE(benjamin): Pinned because onnxruntime changed the way quantization is done, and we need to update our code to support it
78
onnxruntime<1.16
89
# NOTE(alan): Pinned because this is when the most recent module we import appeared

0 commit comments

Comments
 (0)