|
imports: |
|
- $import os |
|
- $import glob |
|
- $import torch |
|
- $import scripts |
|
- $import scripts.inference |
|
|
|
device: '$torch.device("cuda" if torch.cuda.is_available() else "cpu")' |
|
|
|
model_config: |
|
_target_: scripts.exaonepath.EXAONEPathV1Downstream |
|
step_size: 256 |
|
patch_size: 256 |
|
macenko: true |
|
device: '$@device' |
|
|
|
model: '$@model_config' |
|
|
|
input_dir: 'samples' |
|
input_files: '$sorted(glob.glob(@input_dir+''/*.svs''))' |
|
|
|
root_dir: '$os.path.dirname(os.path.dirname(scripts.__file__))' |
|
|
|
inference: |
|
- [email protected]_state_dict(torch.load(os.path.join(@root_dir, "models/exaonepath_v1.0.0_msi.pt"))) |
|
- $scripts.inference.infer(@model, @input_files) |
|
|