提交 98c80f16 编写于 作者: X xxxpsyduck

revert change in db_process.py

上级 72b453b9
......@@ -104,7 +104,7 @@ class DBProcessTrain(object):
if imgvalue is None:
logger.info("{} does not exist!".format(img_path))
return None
elif len(list(imgvalue.shape)) == 2 or imgvalue.shape[2] == 1:
if len(list(imgvalue.shape)) == 2 or imgvalue.shape[2] == 1:
imgvalue = cv2.cvtColor(imgvalue, cv2.COLOR_GRAY2BGR)
data = self.make_data_dict(imgvalue, gt_label)
data = AugmentData(data)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册