12345678910111213141516 |
- import fastdeploy as fd
- from sx_utils.sxtime import sxtimeit
- d_option = fd.RuntimeOption()
- d_option.use_cpu()
- d_model = fd.vision.detection.YOLOv3('./d_models/model.pdmodel',
- "./d_models/model.pdiparams",
- "./d_models/infer_cfg.yml",
- runtime_option=d_option)
- @sxtimeit
- def d_stamp(img):
- result = d_model.predict(img.copy())
- print(result.scores)
- return next(('True' for i in result.scores if i > 0.1), 'False')
|