mirror of
https://github.com/leigest519/ScreenCoder.git
synced 2026-02-13 02:02:48 +00:00
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
import multiprocessing
|
|
from glob import glob
|
|
import time
|
|
import json
|
|
from tqdm import tqdm
|
|
from os.path import join as pjoin, exists
|
|
|
|
import merge
|
|
|
|
|
|
input_root = 'E:\\Mulong\\Datasets\\rico\\combined'
|
|
output_root = 'E:\\Mulong\\Result\\rico\\rico_uied\\rico_new_uied_cls\\merge'
|
|
compo_root = 'E:\\Mulong\\Result\\rico\\rico_uied\\rico_new_uied_cls\\ip'
|
|
text_root = 'E:\\Mulong\\Result\\east'
|
|
|
|
data = json.load(open('E:\\Mulong\\Datasets\\rico\\instances_test.json', 'r'))
|
|
input_paths_img = [pjoin(input_root, img['file_name'].split('/')[-1]) for img in data['images']]
|
|
input_paths_img = sorted(input_paths_img, key=lambda x: int(x.split('\\')[-1][:-4])) # sorted by index
|
|
|
|
# set the range of target inputs' indices
|
|
num = 0
|
|
start_index = 0
|
|
end_index = 100000
|
|
for input_path_img in input_paths_img:
|
|
index = input_path_img.split('\\')[-1][:-4]
|
|
if int(index) < start_index:
|
|
continue
|
|
if int(index) > end_index:
|
|
break
|
|
|
|
merge.incorporate(input_path_img, compo_root, text_root, output_root, resize_by_height=800, show=False)
|