Skip to content

Commit

Permalink
Added default zscore values for training
Browse files Browse the repository at this point in the history
  • Loading branch information
Karol-G committed Aug 9, 2023
1 parent e7b81fc commit aceada4
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion particleseg3d/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.2.8"
__version__ = "0.2.9"
4 changes: 2 additions & 2 deletions particleseg3d/train/preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@ def preprocess_all(load_dir: str, names: List[str], save_dir: str, target_spacin
:param zscore: The z-score used for intensity normalization.
"""
metadata_load_filepath = join(load_dir, "metadata.json")
zscore_load_filepath = join(load_dir, "zscore.json")

with open(metadata_load_filepath) as f:
metadata = json.load(f)
Expand Down Expand Up @@ -205,9 +204,10 @@ def main():
parser.add_argument('-i', "--input", required=True,
help="Absolute input path to the base folder that contains the dataset structured in the form of the directories 'images' and 'instance_seg' and the file metadata.json.")
parser.add_argument('-o', "--output", required=True, help="Absolute output path to the preprocessed dataset directory.")
parser.add_argument('-z', '--zscore', required=True, type=float, nargs=2, help="The z-score used for intensity normalization.")
parser.add_argument('-n', "--name", required=False, type=str, default=None, nargs="+", help="(Optional) The name(s) without extension of the image(s) that should be used for training. Multiple names must be separated by spaces.")
parser.add_argument('-t', '--task', required=False, default=500, type=int, help="(Optional) The task id that should be assigned to this dataset.")
parser.add_argument('-z', '--zscore', default=(5850.29762143569, 7078.294543817302), required=False, type=float, nargs=2,
help="(Optional) The target spacing in millimeters given as three numbers separate by spaces.")
parser.add_argument('-target_particle_size', default=60, required=False, type=int,
help="(Optional) The target particle size in pixels given as three numbers separate by spaces.")
parser.add_argument('-target_spacing', default=0.1, required=False, type=float,
Expand Down

0 comments on commit aceada4

Please sign in to comment.