mirror of
https://github.com/cdtools-developers/cdtools.git
synced 2026-09-18 16:42:09 +02:00
34 lines
797 B
Python
34 lines
797 B
Python
from __future__ import division, print_function, absolute_import
|
|
|
|
import CDTools
|
|
from CDTools.tools import cmath
|
|
from CDTools.tools.plotting import *
|
|
import h5py
|
|
import torch as t
|
|
import numpy as np
|
|
from matplotlib import pyplot as plt
|
|
import pickle
|
|
|
|
|
|
filename = 'example_data/AuBalls_700ms_30nmStep_3_6SS_filter.cxi'
|
|
|
|
with h5py.File(filename,'r') as f:
|
|
dataset = CDTools.datasets.Ptycho_2D_Dataset.from_cxi(f)
|
|
|
|
|
|
model = CDTools.models.SimplePtycho.from_dataset(dataset)
|
|
|
|
for i, loss in enumerate(model.Adam_optimize(10, dataset)):
|
|
print(i, loss)
|
|
|
|
for i, loss in enumerate(model.Adam_optimize(10, dataset, lr=0.001)):
|
|
print(i, loss)
|
|
|
|
model.inspect(update=False)
|
|
model.compare(dataset)
|
|
|
|
#with open('test_results.pickle', 'wb') as f:
|
|
# pickle.dump(model.save_results(),f)
|
|
|
|
plt.show()
|