topaz.protocols.protocol_topaz_picking module¶
- class topaz.protocols.protocol_topaz_picking.TopazProtPicking(**args)[source]¶
Bases:
pwem.protocols.protocol_particles_picking.ProtParticlePickingAuto
,topaz.protocols.protocol_base.ProtTopazBase
Perform a picking using a topaz model
- ADD_MODEL_GENERAL = 1¶
- ADD_MODEL_PRETRAINED = 0¶
- ADD_MODEL_TRAIN_TYPES = ['TopazTrained', 'TopazGeneral']¶
- GENERAL_MODELS = ['resnet16_u64', 'resnet16_u32', 'resnet8_u64', 'resnet8_u32']¶
- MODEL_RESNET16_U32 = 1¶
- MODEL_RESNET16_U64 = 0¶
- MODEL_RESNET8_U32 = 3¶
- MODEL_RESNET8_U64 = 2¶