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