24 lines
806 B
Python
24 lines
806 B
Python
from os import listdir
|
|
import subprocess
|
|
from tqdm import tqdm
|
|
import re
|
|
IMG_PATH="./../../import/images/"
|
|
files=listdir(IMG_PATH)
|
|
all=[]
|
|
for file in tqdm(files):
|
|
file_path=f"{IMG_PATH}/{file}"
|
|
result = subprocess.run(f"resdet {file_path}",shell=True, stdout=subprocess.PIPE)
|
|
res_str=result.stdout.decode("utf-8").splitlines()
|
|
if len(res_str)<2:
|
|
continue
|
|
if "(not upsampled)" in res_str[1]:
|
|
continue
|
|
given_resolution=re.findall(r'\d+', res_str[0])
|
|
given_pixels_total=int(given_resolution[0])*int(given_resolution[1])
|
|
best_guess_resolution=re.findall(r'\d+', res_str[1])
|
|
best_pixels_total=int(best_guess_resolution[0])*int(best_guess_resolution[1])
|
|
if best_pixels_total/given_pixels_total<0.6:
|
|
all.append(file_path)
|
|
print(all)
|
|
print(len(all))
|