import numpy as np | |
import torch | |
import shutil | |
import os | |
import matplotlib.pyplot as plt | |
import cv2 | |
import json | |
from PIL import Image | |
import pickle | |
from skimage.transform import resize | |
from utils.dataset_prepare import split_data, save_fileLabel | |
def Kvasir_split(): | |
dataset_name = "Kvasir" | |
file = "./dataset_demo/Kvasir" | |
if os.path.exists(os.path.join(file, "masks")): | |
os.rename(os.path.join(file, "masks"), os.path.join(file, "labels")) | |
split_data(dataset_name) | |
save_fileLabel(dataset_name) |