# import packagesimport numpy as npimport osfrom glob import globimport pandas as pdjoin = ospathjoinfrom skimage import transform io segmentationfrom tqdm import tqdmimport torchfrom segment_anythin
输入地址:
img_path: 原始图像所在文件夹的路径。gt_path: ground truth(标注图像)所在文件夹的路径。csv: 以CSV格式给出所有图像和标注图像的文件名和路径的CSV文件的路径。如果不提供CSV文件,则程序将在img_path和gt_path中查找所有图像和标注图像的文件名。checkpoint: 预训练模型的路径。
输出地址:
npz_path: 将预处理后的图像和标注保存为.npz格式的文件所在文件夹的路径。每个.npz文件包含三个 numpy 数组:imgs、gts和img_embeddings。save_path: 在npz_path后添加_model_type后缀的路径,用于保存使用不同模型预处理得到的.npz文件。data_name:.npz文件的名称,用于标识数据集名称。save_path + ".png": 为了检查预处理的结果是否正确,程序会随机选择一个图像,并将其保存为.png格式的文件
原文地址: https://www.cveoy.top/t/topic/fnX5 著作权归作者所有。请勿转载和采集!