(self)
| 339 | return OrderedDict([('lr_G', self._current_lr_G), ('lr_D', self._current_lr_D)]) |
| 340 | |
| 341 | def get_current_visuals(self): |
| 342 | # visuals return dictionary |
| 343 | visuals = OrderedDict() |
| 344 | |
| 345 | # input visuals |
| 346 | title_input_img = os.path.basename(self._input_real_img_path[0]) |
| 347 | visuals['1_input_img'] = plot_utils.plot_au(self._vis_real_img, self._vis_real_cond, title=title_input_img) |
| 348 | visuals['2_fake_img'] = plot_utils.plot_au(self._vis_fake_img, self._vis_desired_cond) |
| 349 | visuals['3_rec_real_img'] = plot_utils.plot_au(self._vis_rec_real_img, self._vis_real_cond) |
| 350 | visuals['4_fake_img_unmasked'] = self._vis_fake_img_unmasked |
| 351 | visuals['5_fake_img_mask'] = self._vis_fake_img_mask |
| 352 | visuals['6_rec_real_img_mask'] = self._vis_rec_real_img_mask |
| 353 | visuals['7_cyc_img_unmasked'] = self._vis_fake_img_unmasked |
| 354 | # visuals['8_fake_img_mask_sat'] = self._vis_fake_img_mask_saturated |
| 355 | # visuals['9_rec_real_img_mask_sat'] = self._vis_rec_real_img_mask_saturated |
| 356 | visuals['10_batch_real_img'] = self._vis_batch_real_img |
| 357 | visuals['11_batch_fake_img'] = self._vis_batch_fake_img |
| 358 | visuals['12_batch_fake_img_mask'] = self._vis_batch_fake_img_mask |
| 359 | # visuals['11_idt_img'] = self._vis_idt_img |
| 360 | |
| 361 | return visuals |
| 362 | |
| 363 | def save(self, label): |
| 364 | # save networks |
nothing calls this directly
no outgoing calls
no test coverage detected