提交 b8b7ae40 编写于 作者: S stephon

rm useless info

上级 cd9b8cad
......@@ -37,7 +37,6 @@ class ImagenetOp(Op):
label_idx += 1
def preprocess(self, input_dicts, data_id, log_id):
print("111111")
(_, input_dict), = input_dicts.items()
batch_size = len(input_dict.keys())
imgs = []
......@@ -48,11 +47,9 @@ class ImagenetOp(Op):
img = self.seq(im)
imgs.append(img[np.newaxis, :].copy())
input_imgs = np.concatenate(imgs, axis=0)
print("2222222")
return {"inputs": input_imgs}, False, None, ""
def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
print("3333333")
score_list = fetch_dict["prediction"]
result = {"label": [], "prob": []}
for score in score_list:
......@@ -63,7 +60,6 @@ class ImagenetOp(Op):
result["prob"].append(max_score)
result["label"] = str(result["label"])
result["prob"] = str(result["prob"])
print("444444444")
return result, None, ""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册