Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Sep 23, 2024
1 parent f1a3af4 commit 501911c
Showing 1 changed file with 34 additions and 34 deletions.
68 changes: 34 additions & 34 deletions models/vista3d/configs/inference.json
Original file line number Diff line number Diff line change
Expand Up @@ -138,40 +138,40 @@
"sw_batch_size": "@sw_batch_size",
"use_point_window": "@use_point_window"
},
"postprocessing_transforms":[
{
"_target_": "ToDeviced",
"keys": "pred",
"device": "cpu",
"_disabled_": true
},
{
"_target_": "monai.apps.vista3d.transforms.VistaPostTransformd",
"keys": "pred"
},
{
"_target_": "Invertd",
"keys": "pred",
"transform": "$copy.deepcopy(@preprocessing)",
"orig_keys": "@image_key",
"nearest_interp": true,
"to_tensor": true
},
{
"_target_": "Lambdad",
"func": "$lambda x: torch.nan_to_num(x, nan=255)",
"keys": "pred"
},
{
"_target_": "SaveImaged",
"keys": "pred",
"resample": false,
"output_dir": "@output_dir",
"output_ext": "@output_ext",
"output_dtype": "@output_dtype",
"output_postfix": "@output_postfix",
"separate_folder": "@separate_folder"
}
"postprocessing_transforms": [
{
"_target_": "ToDeviced",
"keys": "pred",
"device": "cpu",
"_disabled_": true
},
{
"_target_": "monai.apps.vista3d.transforms.VistaPostTransformd",
"keys": "pred"
},
{
"_target_": "Invertd",
"keys": "pred",
"transform": "$copy.deepcopy(@preprocessing)",
"orig_keys": "@image_key",
"nearest_interp": true,
"to_tensor": true
},
{
"_target_": "Lambdad",
"func": "$lambda x: torch.nan_to_num(x, nan=255)",
"keys": "pred"
},
{
"_target_": "SaveImaged",
"keys": "pred",
"resample": false,
"output_dir": "@output_dir",
"output_ext": "@output_ext",
"output_dtype": "@output_dtype",
"output_postfix": "@output_postfix",
"separate_folder": "@separate_folder"
}
],
"postprocessing": {
"_target_": "Compose",
Expand Down

0 comments on commit 501911c

Please sign in to comment.