Skip to content

Commit a74f738

Browse files
Add mask for handwritten text
1 parent 4582945 commit a74f738

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

trdg/data_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def generate(
6060
if is_handwritten:
6161
if orientation == 1:
6262
raise ValueError("Vertical handwritten text is unavailable")
63-
image = handwritten_text_generator.generate(text, text_color)
63+
image, mask = handwritten_text_generator.generate(text, text_color)
6464
else:
6565
image, mask = computer_text_generator.generate(
6666
text,

0 commit comments

Comments
 (0)