debug: should not use == to check None obj

pull/2502/head
gaotingquan 2022-11-16 09:13:23 +00:00 committed by Tingquan Gao
parent 89696c7bac
commit 012823bed5
1 changed files with 1 additions and 1 deletions

View File

@ -723,7 +723,7 @@ class PaddleClas(object):
prediction result(s) is zipped as a dict, that includs topk "class_ids", "scores" and "label_names".
The format of batch prediction result(s) is as follow: [{"class_ids": [...], "scores": [...], "label_names": [...]}, ...]
"""
if input_data == None and self._config.Global.infer_imgs:
if input_data is None and self._config.Global.infer_imgs:
input_data = self._config.Global.infer_imgs
if isinstance(input_data, np.ndarray):