[ILUVATAR_GPU] Support for iluvatar_gpu

Author: tianyuzhou668

ppocr/data/imaug/operators.py

@@ -230,7 +230,7 @@ def __call__(self, data):
             # img, shape = self.resize_image_type1(img)
             img, [ratio_h, ratio_w] = self.resize_image_type1(img)
         data["image"] = img
-        data["shape"] = np.array([src_h, src_w, ratio_h, ratio_w])
+        data["shape"] = np.array([src_h, src_w, ratio_h, ratio_w]).astype(np.float32)
         return data
 
     def image_padding(self, im, value=0):

ppocr/data/imaug/random_crop_data.py

@@ -178,7 +178,7 @@ def __call__(self, data):
                 ignore_tags_crop.append(tag)
                 texts_crop.append(text)
         data["image"] = img
-        data["polys"] = np.array(text_polys_crop)
+        data["polys"] = np.array(text_polys_crop).astype(np.float32)
         data["ignore_tags"] = ignore_tags_crop
         data["texts"] = texts_crop
         return data

ppocr/data/imaug/rec_img_aug.py

@@ -304,7 +304,7 @@ def __call__(self, data):
         else:
             norm_img, valid_ratio = resize_norm_img(img, self.image_shape, self.padding)
         data["image"] = norm_img
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -337,7 +337,7 @@ def __call__(self, data):
         valid_ratio = min(1.0, float(resized_w / imgW))
 
         data["image"] = norm_img
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -365,7 +365,7 @@ def __call__(self, data):
             img, self.image_shape, self.padding, self.interpolation
         )
         data["image"] = norm_img
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -406,7 +406,7 @@ def __call__(self, data):
         data["image"] = norm_img
         data["resized_shape"] = resize_shape
         data["pad_shape"] = pad_shape
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -538,7 +538,7 @@ def __call__(self, data):
         img = data["image"]
         norm_img, valid_ratio = resize_norm_img_abinet(img, self.image_shape)
         data["image"] = norm_img
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -552,7 +552,7 @@ def __call__(self, data):
 
         norm_img, valid_ratio = resize_norm_img(img, self.image_shape, self.padding)
         data["image"] = norm_img
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
 
@@ -573,7 +573,7 @@ def __call__(self, data):
         data["image"] = norm_img
         data["resized_shape"] = resize_shape
         data["pad_shape"] = pad_shape
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         data["word_positons"] = word_positons
         return data
 

ppocr/data/simple_dataset.py

@@ -202,7 +202,7 @@ def resize_norm_img(self, data, imgW, imgH, padding=True):
         padding_im[:, :, :resized_w] = resized_image
         valid_ratio = min(1.0, float(resized_w / imgW))
         data["image"] = padding_im
-        data["valid_ratio"] = valid_ratio
+        data["valid_ratio"] = np.float32(valid_ratio)
         return data
 
     def __getitem__(self, properties):

tools/program.py

@@ -115,7 +115,7 @@ def merge_config(config, opts):
     return config
 
 
-def check_device(use_gpu, use_xpu=False, use_npu=False, use_mlu=False, use_gcu=False):
+def check_device(use_gpu, use_xpu=False, use_npu=False, use_mlu=False, use_gcu=False, use_iluvatar_gpu=False):
     """
     Log error and exit when set use_gpu=true in paddlepaddle
     cpu version.
@@ -833,6 +833,7 @@ def preprocess(is_train=False):
     use_npu = config["Global"].get("use_npu", False)
     use_mlu = config["Global"].get("use_mlu", False)
     use_gcu = config["Global"].get("use_gcu", False)
+    use_iluvatar_gpu = config["Global"].get("use_iluvatar_gpu", False)
 
     alg = config["Architecture"]["algorithm"]
     assert alg in [
@@ -896,9 +897,11 @@ def preprocess(is_train=False):
         device = "mlu:{0}".format(os.getenv("FLAGS_selected_mlus", 0))
     elif use_gcu:  # Use Enflame GCU(General Compute Unit)
         device = "gcu:{0}".format(os.getenv("FLAGS_selected_gcus", 0))
+    elif use_iluvatar_gpu:
+        device = "iluvatar_gpu:{0}".format(dist.ParallelEnv().dev_id)
     else:
         device = "gpu:{}".format(dist.ParallelEnv().dev_id) if use_gpu else "cpu"
-    check_device(use_gpu, use_xpu, use_npu, use_mlu, use_gcu)
+    check_device(use_gpu, use_xpu, use_npu, use_mlu, use_gcu, use_iluvatar_gpu)
 
     device = paddle.set_device(device)