From a814a6fa66903a0dec44f5db11ca2c8ef71bd4af Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Fri, 31 May 2024 17:30:47 +0800 Subject: [PATCH] Update utils.py --- utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/utils.py b/utils/utils.py index 94b903b..e4eaa99 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -4,6 +4,7 @@ import numpy as np from PIL import Image,ImageOps,ImageDraw MAX_COLORS = 12 import os +import random dir_path = os.path.dirname(os.path.abspath(__file__)) path_dir = os.path.dirname(dir_path) file_path = os.path.dirname(path_dir)