plt原图像素+去四周白边和坐标轴保存图片

在复现一个floor_plan的语义分割的任务的时候想提取其中的一个子任务，也就是平面图识别中的房间识别，于是我需要把标记好的图片通过plt保存，并且维持原分辨率（像素）并且去除坐标轴和白边，看了很多教程，探索出了一段可以批量处理Cubicasa数据集房间识别的代码：

import cv2
import shutil
import numpy as np
import matplotlib.pyplot as plt
from tqdm import tqdm
from floortrans.loaders import FloorplanSVG
from torch.utils.data import DataLoader
from floortrans.plotting import discrete_cmap

discrete_cmap()

room_classes = ["Background", "Outdoor", "Wall", "Kitchen", "Living Room" ,"Bed Room", "Bath", "Entry", "Railing", "Storage", "Garage", "Undefined"] #不同的房间类型
data_folder = 'data/cubicasa140/' # 数据集文件夹
data_file = 'train.txt' # 数据集类型
savedata_folder = 'data_/pic/'
savelabel_folder = 'data_/label/' 

normal_set = FloorplanSVG(data_f