diff --git a/tools/data/textdet/totaltext_converter.py b/tools/data/textdet/totaltext_converter.py
index 3ad578ba..abe4739c 100644
--- a/tools/data/textdet/totaltext_converter.py
+++ b/tools/data/textdet/totaltext_converter.py
@@ -180,7 +180,7 @@ def process_line(line, contours, words):
     ann_dict = re.sub('([0-9]) +([ 0-9])', r'\1,\2', ann_dict)
     ann_dict = re.sub('([0-9]) -([0-9])', r'\1,-\2', ann_dict)
     ann_dict = ann_dict.replace("[u',']", "[u'#']")
-    ann_dict = yaml.load(ann_dict)
+    ann_dict = yaml.safe_load(ann_dict)
 
     X = np.array([ann_dict['x']])
     Y = np.array([ann_dict['y']])
diff --git a/tools/data/textrecog/totaltext_converter.py b/tools/data/textrecog/totaltext_converter.py
index bd473037..38b5b9f4 100644
--- a/tools/data/textrecog/totaltext_converter.py
+++ b/tools/data/textrecog/totaltext_converter.py
@@ -169,7 +169,7 @@ def process_line(line, contours, words):
     ann_dict = re.sub('([0-9]) +([ 0-9])', r'\1,\2', ann_dict)
     ann_dict = re.sub('([0-9]) -([0-9])', r'\1,-\2', ann_dict)
     ann_dict = ann_dict.replace("[u',']", "[u'#']")
-    ann_dict = yaml.load(ann_dict)
+    ann_dict = yaml.safe_load(ann_dict)
 
     X = np.array([ann_dict['x']])
     Y = np.array([ann_dict['y']])