浏览代码

Merge pull request #705 from FlyingQianMM/develop_qh

fix 'local variable 'im' referenced before assignment' for classification transform
FlyingQianMM 4 年之前
父节点
当前提交
cb8c045504
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      paddlex/cv/transforms/cls_transforms.py

+ 1 - 0
paddlex/cv/transforms/cls_transforms.py

@@ -73,6 +73,7 @@ class Compose(ClsTransform):
                 raise Exception(
                     "im should be 3-dimension, but now is {}-dimensions".
                     format(len(im_file.shape)))
+            im = im_file
         else:
             try:
                 if input_channel == 3: