Source code for cryocare.objects


import pyworkflow.object as pwobj
from pwem import EMObject


[docs]class CryocareTrainData(EMObject): def __init__(self, train_data=None, mean_std=None, patch_size=None, **kwargs): EMObject.__init__(self, **kwargs) self._train_data = pwobj.String(train_data) self._mean_std = pwobj.String(mean_std) self._patch_size = pwobj.Integer(patch_size)
[docs] def getTrainData(self): return self._train_data.get()
[docs] def getMeanStd(self): return self._mean_std.get()
[docs] def getPatchSize(self): return self._patch_size.get()
[docs] def getPath(self): return '{}\n{}'.format(self.getTrainData(), self.getMeanStd())
def __str__(self): return "CryoCARE Train Data (path=%s)" % self.getPath()
[docs]class CryocareModel(EMObject): def __init__(self, basedir=None, mean_std=None, **kwargs): EMObject.__init__(self, **kwargs) self._basedir = pwobj.String(basedir) self._mean_std = pwobj.String(mean_std)
[docs] def getPath(self): return self._basedir.get()
[docs] def getMeanStd(self): return self._mean_std.get()
def __str__(self): return "CryoCARE Model (path=%s)" % self.getPath()