Describe the bug
In both the N2VModule and the CAREModule the additional_metadata field in ImageRegionData is getting replaced with an empty dict in predict_step. Currently this field contains the zarr chunk size and shard size which means this won't be preserved when writing predictions to zarr.
Describe the bug
In both the
N2VModuleand theCAREModuletheadditional_metadatafield inImageRegionDatais getting replaced with an empty dict inpredict_step. Currently this field contains the zarr chunk size and shard size which means this won't be preserved when writing predictions to zarr.careamics/src/careamics/lightning/modules/n2v_module.py
Line 188 in a9033f5