cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-18 08:42:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix the initialization issue with nonuniform intensities

Browse Source
This commit is contained in:
levita_a
2025-07-21 17:00:43 +02:00
parent 22a28f4043
commit 2a00ba9c21
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/cdtools/models/fancy_ptycho.py
+3 -1
View File
@@ -407,7 +407,9 @@ class FancyPtycho(CDIModel):
if hasattr(dataset, 'intensities') and dataset.intensities is not None:
intensities = dataset.intensities.to(dtype=Ws.dtype)[:,...]
weights = t.sqrt(intensities)
Ws *= (weights / t.mean(weights))
Ws *= (weights / t.mean(weights)).reshape(
(len(weights),) + (1,)*(Ws.ndim - 1))
if hasattr(dataset, 'mask') and dataset.mask is not None:
mask = dataset.mask.to(t.bool)