STDC/shiyan.py

import os
import time
from PIL import Image
import numpy as np
from models_725.segWaterBuilding import SegModel
from predict_guolv import predict_lunkuo


impth = '/home/data/lijiwen/wurenjiqifei/images/'
folders = os.listdir(impth)
segmodel = SegModel()
for i in range(len(folders)):
    imgpath = os.path.join(impth, folders[i])
    # name = imgpath.split('/')[-1].split('.')[0]
    time00 = time.time()
    img = Image.open(imgpath).convert('RGB')
    img = np.array(img)
    time11 = time.time()
    img_out = predict_lunkuo(impth=img, segmodel=segmodel, filter=False)
    print('----all_process', (time.time() - time11) * 1000)