super-gradients icon indicating copy to clipboard operation
super-gradients copied to clipboard

Predict_webcam is not working for yolo nas pose

Open Daanfb opened this issue 1 year ago • 4 comments

🐛 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

Daanfb avatar Feb 15 '24 07:02 Daanfb