cSAXS/cdtools
0
0 Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-11-14 10:13:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor datasets, wrote more docs, added one more test dataset

Browse Source
This commit is contained in:
Abe Levitan committed 2019-09-09 14:16:42 -04:00
1 parent 99d02f95cf
commit 40998908f1
20 files changed
+338 -160
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

CDTools/models/fancy_ptycho.py
+2 -2
View File
@@ -2,7 +2,7 @@ from __future__ import division, print_function, absolute_import
import torch as t
from CDTools.models import CDIModel
from CDTools.datasets import Ptycho_2D_Dataset
from CDTools.datasets import Ptycho2DDataset
from CDTools import tools
from CDTools.tools import cmath
from CDTools.tools import plotting as p
@@ -316,7 +316,7 @@ class FancyPtycho(CDIModel):
data = self.forward(indices, translations)
# And finally, we make the dataset
return Ptycho_2D_Dataset(translations, data,
return Ptycho2DDataset(translations, data,
entry_info = entry_info,
sample_info = sample_info,
wavelength=wavelength,
CDTools/models/pinhole_plane_ptycho.py
+2 -2
View File
@@ -2,7 +2,7 @@ from __future__ import division, print_function, absolute_import
import torch as t
from CDTools.models import CDIModel
from CDTools.datasets import Ptycho_2D_Dataset
from CDTools.datasets import Ptycho2DDataset
from CDTools import tools
from CDTools.tools import cmath
from CDTools.tools import plotting as p
@@ -331,7 +331,7 @@ class PinholePlanePtycho(CDIModel):
data = self.forward(indices, translations)
# And finally, we make the dataset
return Ptycho_2D_Dataset(translations, data,
return Ptycho2DDataset(translations, data,
entry_info = entry_info,
sample_info = sample_info,
wavelength=wavelength,
CDTools/models/simple_ptycho.py
+2 -2
View File
@@ -2,7 +2,7 @@ from __future__ import division, print_function, absolute_import
import torch as t
from CDTools.models import CDIModel
from CDTools.datasets import Ptycho_2D_Dataset
from CDTools.datasets import Ptycho2DDataset
from CDTools import tools
from CDTools.tools import plotting as p
from copy import copy
@@ -177,7 +177,7 @@ class SimplePtycho(CDIModel):
data = self.forward(indices, translations)
# And finally, we make the dataset
return Ptycho_2D_Dataset(translations, data,
return Ptycho2DDataset(translations, data,
entry_info = entry_info,
sample_info = sample_info,
wavelength=wavelength,