|
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) |
|
|
|
|