Adds two classmethods to CDIModel and its ptychography subclasses that
allow models to be reconstructed from saved .h5 result files without
needing the original dataset.
- CDIModel base class: add model_class and cdtools_version to
save_results(); add _load_results_dict() helper (restores state_dict
+ training metadata); add from_results_dict() interface method and a
concrete from_results_h5() that reads a file and delegates to
from_results_dict() — subclasses inherit this for free.
- SimplePtycho: make save_results(dataset=None) optional (dataset was
accepted but never used); add from_results_dict() which reconstructs
from probe, obj, wavelength, probe_basis, and min_translation stored
in the state_dict.
- FancyPtycho: add optional translations parameter to __init__,
registered as original_translations buffer; update from_dataset to
pass translations; make corrected_translations(dataset=None) fall back
to self.original_translations when no dataset is provided; make
save_results(dataset=None) use stored translations when no dataset is
provided; add from_results_dict() that detects all optional features
(mask, translation_offsets, weights, near-field propagators, etc.)
from the state_dict and reconstructs the full model. Also fix a latent
bug where background was passed to t.nn.Parameter() without
t.as_tensor(), which now fails when given a numpy array.
All existing tests pass; new tests added for both models verifying that
state_dict, training metadata, and forward pass output are all restored
exactly after a round-trip through from_results_dict and from_results_h5.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
- Revert test_Adam_gold_balls to default amplitude_mse
- Add new test_intensity_MSE test (gold balls + AdamReconstructor)
- Update near_field threshold to 17 (poisson_nll scale)
Thresholds for test_intensity_MSE to be tuned after GPU run.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Use poisson_nll in test_near_field_ptycho and intensity_mse in
test_Adam_gold_balls to exercise these loss paths end-to-end.
Thresholds left as-is pending re-running on a GPU machine.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
poisson_nll now returns a sum rather than a mean, consistent with the
normalizer pattern. Remove the per-pixel divisions from the numpy
reference calculations accordingly.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
* Added a function to allow for deletion of translation positions via logical indexing
In ptycho_2d_dataset.py
- Created remove_translations_mask. This handles logical-indexing-based deletion of translation points
- Refactored crop_translations to be dependent on remove_translations_mask
* Add validation and tests for remove_translations_mask method in Ptycho2DDataset
* Switch mask to mask_remove
---------
Co-authored-by: gnzng <damian@guenzing.de>
Co-authored-by: Clemens Schmid <clemisch@tuta.io>
For crop_translations in ptycho_2d_dataset.py
- crop_2d_translations is now crop_translations
- roi is now order-insensitive within the first and last pairs of elements.
- the definition of top, bottom, left, and right, are now consistent with figures generated by matplotlib.pyplot.imshow
- self.intensities is now updated
For tests_datasets.py
- the order-insensitivity of roi in crop_translation is tested, and tests for specific ordering of values in roi are removed.
levita_a
allevitan