Esempi in Python per prepare_training_evolve

Esempio n. 1

File: evolve.py Progetto: DlutMedimgGroup/AnatomySketch-Software

 def prepare_training_evolve(self, output, batch, init):
     evolve = snake_gcn_utils.prepare_training_evolve(
         output['ex_pred'], init)
     output.update({
         'i_it_py': evolve['i_it_py'],
         'c_it_py': evolve['c_it_py'],
         'i_gt_py': evolve['i_gt_py']
     })
     evolve.update({'py_ind': init['py_ind']})
     return evolve

Esempio n. 2

 def prepare_training_evolve(self, output, batch, init):
     ct_num = batch['meta']['ct_num'].sum()
     evolve = snake_gcn_utils.prepare_training_evolve(
         output['ex_pred'], init, ct_num)
     output.update({
         'i_it_py': evolve['i_it_py'],
         'c_it_py': evolve['c_it_py'],
         'i_gt_py': evolve['i_gt_py']
     })
     evolve.update({'ind': init['ind'][:evolve['i_gt_py'].size(0)]})
     return evolve