super-gradients
super-gradients copied to clipboard
Predict_webcam is not working for yolo nas pose
🐛 Describe the bug
I'm trying to get the webcam prediction with YOLO-NAS-POSE and I get an error. My OS is Windows.
My code is the following:
from super_gradients.training import models
yolo_nas_pose_s = models.get("yolo_nas_pose_s", pretrained_weights="coco_pose").cpu()
yolo_nas_pose_s.predict_webcam(conf=0.5)
The error is the following:
File [c:\Users\danie\anaconda3\envs\yolo-nas-pose\lib\site-packages\super_gradients\training\models\detection_models\customizable_detector.py:329](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:329), in CustomizableDetector.predict_webcam(self, iou, conf, fuse_model, skip_image_resizing, nms_top_k, max_predictions, multi_label_per_box, class_agnostic_nms)
[303](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:303) def predict_webcam(
[304](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:304) self,
[305](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:305) iou: Optional[float] = None,
(...)
[312](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:312) class_agnostic_nms: Optional[bool] = None,
[313](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:313) ):
[314](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:314) """Predict using webcam.
[315](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:315)
[316](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:316) :param iou: (Optional) IoU threshold for the nms algorithm. If None, the default value associated to the training is used.
(...)
[327](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:327) If False NMS is performed separately for each class.
[328](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:328) """
--> [329](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:329) pipeline = self._get_pipeline(
[330](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:330) iou=iou,
[331](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:331) conf=conf,
[332](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:332) fuse_model=fuse_model,
[333](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:333) skip_image_resizing=skip_image_resizing,
[334](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:334) nms_top_k=nms_top_k,
[335](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:335) max_predictions=max_predictions,
[336](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:336) multi_label_per_box=multi_label_per_box,
[337](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:337) class_agnostic_nms=class_agnostic_nms,
[338](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:338) )
[339](file:///C:/Users/danie/anaconda3/envs/yolo-nas-pose/lib/site-packages/super_gradients/training/models/detection_models/customizable_detector.py:339) pipeline.predict_webcam()
TypeError: YoloNASPose._get_pipeline() got an unexpected keyword argument 'nms_top_k'
Versions
super_gradients == 3.6.0