allrun_uavid_testacc.py

'''
2024.5.14: test uavid dataset
'''
import os
from pathlib import Path
from glob import glob

SAVE_WEIGHT_PATH = '/localssd/yinxia/CRGNetdata/'

if __name__=="__main__":
    city = 'uavid'
    #############################
    # backbonelist = ['effiSAM', ]
    backbonelist = ['effiSAM', 'unetformer', 'FCNcls']
    # suffix = '_nocrf'
    suffix = '_wcrf'
    for backbone in backbonelist:
        # directly find the 5000.pth
        respath = f'{SAVE_WEIGHT_PATH}/{city}/{backbone}/'
        # print(respath)
        weightlist1 = Path(f'{SAVE_WEIGHT_PATH}/{city}/{backbone}/').rglob('*batch5000*.pth')
        weightlist2 = Path(f'{SAVE_WEIGHT_PATH}/{city}/{backbone}/').rglob('*batch5001*.pth')
        weightlist = [i for i in weightlist1] + [i for i in weightlist2]
        print(len(weightlist))

        # process each
        for file in weightlist:
            dirpath = str(file.parent)
            savetxtfile = glob(os.path.join(dirpath, 'Test*.txt'))
            if len(savetxtfile)!=0:
                continue
            print('process: %s'%dirpath)
            dirpath2 = os.path.basename(dirpath)
            method =dirpath2.split('_')[0]
            os.system(
                f"python all_test_acc{suffix}.py --city {city} --backbone {backbone} --method {method} "
                f"--restore_path {dirpath} --restore_iter 5000")