tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.
Apache License 2.0
3.79k stars 476 forks source link

fix demo.py trying to save inference result when running using camera #151

Open SkRagibIshrakAbid opened 2 months ago

SkRagibIshrakAbid commented 2 months ago

When running demo.py with torch to detect object using camera it keeps on trying to save inference result which results in error.