pull/12/head
d8ahazard 2022-12-10 18:00:02 -06:00
parent bead6660ef
commit cdfc93b8a4
1 changed files with 2 additions and 1 deletions

View File

@ -2,6 +2,7 @@ import math
import os import os
import sys import sys
import traceback import traceback
from pathlib import Path
import numpy as np import numpy as np
import tqdm import tqdm
@ -211,7 +212,7 @@ def preprocess(rename,
return msg, msg return msg, msg
filename = os.path.join(src, src_image) filename = os.path.join(src, src_image)
if not is_image(filename): if not is_image(Path(filename), pil_features):
continue continue
try: try: