AioMedica2 / slide_tiling.py
chris1nexus
First commit
54660f7
import subprocess
import argparse
import os
import shutil
def save_tiles(slide_path):
filename = os.path.basename(slide_path)
FILEID = filename.rsplit('.', maxsplit=1)[0]
PATCHES_DIR = os.environ['PATCHES_DIR']
SLIDES_DIR = os.environ['SLIDES_DIR']
os.makedirs(PATCHES_DIR, exist_ok=True)
os.makedirs(SLIDES_DIR, exist_ok=True)
shutil.copy(slide_path, SLIDES_DIR)
INPUT_PATH = os.path.join(SLIDES_DIR, filename)
CMD = ['python3', 'src/tile_WSI.py', '-s', '512', '-e', '0', '-j', '16', '-B', '50', '-M', '20', '-o', PATCHES_DIR, INPUT_PATH]
subprocess.call(CMD)
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='PyTorch Classification')
parser.add_argument('--slide_path', type=str, help='path to the WSI slide')
args = parser.parse_args()
filename = os.path.basename(args.slide_path)
FILEID = filename.rsplit('.', maxsplit=1)[0]
PATCHES_DIR = os.environ['PATCHES_DIR']
SLIDES_DIR = os.environ['SLIDES_DIR']
os.makedirs(PATCHES_DIR, exist_ok=True)
os.makedirs(SLIDES_DIR, exist_ok=True)
shutil.move(args.slide_path, SLIDES_DIR)
INPUT_PATH = os.path.join(SLIDES_DIR, filename)
CMD = ['python3', 'src/tile_WSI.py', '-s', '512', '-e', '0', '-j', '16', '-B', '50', '-M', '20', '-o', PATCHES_DIR, INPUT_PATH]
subprocess.call(CMD)