atom-predict/msunet/post.py

20 lines
585 B
Python
Executable File

import os
import cv2
import json
import glob
import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
from skimage.measure import label
from skimage.measure import regionprops
from utils.labelme import save_pred_to_json,save_pred_to_json
json_path = './result/predict/test.json'
# save_path = '/home/gao/mouclear/cc/data/detect_50/替换图像'
# json_path = '/home/gao/mouclear/cc/code/pd_2/code/egnn/logs/0/version_2/test.json'
save_path = './result/post_process/'
os.makedirs(save_path, exist_ok=True)
save_pred_to_json(json_path, save_path)