Update ixc_utils.py
#23
by daluggas - opened
- ixc_utils.py +9 -3
ixc_utils.py
CHANGED
|
@@ -9,9 +9,15 @@ import torchvision.transforms as transforms
|
|
| 9 |
from decord import VideoReader
|
| 10 |
|
| 11 |
def get_font():
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
return font
|
| 16 |
|
| 17 |
def padding_336(b, pad=336):
|
|
|
|
| 9 |
from decord import VideoReader
|
| 10 |
|
| 11 |
def get_font():
|
| 12 |
+
current_dir = os.path.dirname(os.path.abspath(__file__))
|
| 13 |
+
font_path = os.path.join(current_dir, 'SimHei.ttf')
|
| 14 |
+
|
| 15 |
+
if os.path.isfile(font_filename):
|
| 16 |
+
font = ImageFont.truetype(font_path, size=40)
|
| 17 |
+
else:
|
| 18 |
+
truetype_url = 'https://huggingface.co/internlm/internlm-xcomposer2d5-7b/resolve/main/SimHei.ttf?download=true'
|
| 19 |
+
ff = urlopen(truetype_url)
|
| 20 |
+
font = ImageFont.truetype(ff, size=40)
|
| 21 |
return font
|
| 22 |
|
| 23 |
def padding_336(b, pad=336):
|