Dur
parent
bead6660ef
commit
cdfc93b8a4
|
|
@ -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:
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue