File size: 519 Bytes
be2715b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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) |