修改测试

This commit is contained in:
lee
2025-07-01 13:39:34 +08:00
parent 96a9445761
commit ed6a4144fa
8 changed files with 55 additions and 14 deletions

View File

@ -253,6 +253,8 @@ class BasePredictor:
# Postprocess
with profilers[2]:
self.results = self.postprocess(preds, im, im0s)
if len(self.results) == 0:
continue
self.run_callbacks("on_predict_postprocess_end")
# Visualize, save, write results