along/semantic-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修bug

Browse Source
This commit is contained in:
weixin_46229132 2025-05-30 09:37:22 +08:00
parent 388f5ed277
commit 8148a58584
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
datasets_pro_code/png_mask_trans.py
View File

@ -16,7 +16,7 @@ for file in os.listdir(src_dir):
mask = mask.copy()
mask[mask == 15] = 255
png_path = os.path.join(dst_dir, os.path.splitext(file)[0] + ".png")
png_path = os.path.join(dst_dir, os.path.splitext(file)[0].replace("mask", "img") + ".png")
Image.fromarray(mask.astype(np.uint8)).save(png_path)
print(f"已保存: {png_path}")