Просмотр исходного кода

fix image feature module order assert (#2771)

cuicheng01 10 месяцев назад
Родитель
Сommit
d9f4870a0d
1 измененных файлов с 0 добавлено и 1 удалено
  1. 0 1
      paddlex/inference/models_new/image_feature/predictor.py

+ 0 - 1
paddlex/inference/models_new/image_feature/predictor.py

@@ -144,7 +144,6 @@ class ImageFeaturePredictor(BasicPredictor):
         channel_num=3,
     ):
         assert channel_num == 3
-        assert order == "hwc"
         return "Normalize", Normalize(scale=scale, mean=mean, std=std)
 
     @register("ToCHWImage")