diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 16:00:32 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 16:00:32 +0800 |
| commit | 2a9507204ae2dd14504556ab5885c4f39bddd89a (patch) | |
| tree | 112351f0844b4a7af24d92a91b327f2fa74f1b16 /models/model.py | |
| parent | 641d19c1ebdf44486de139fadeff3276aecdf284 (diff) | |
Decode mean appearance feature
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/models/model.py b/models/model.py index 0c3e5eb..139fd59 100644 --- a/models/model.py +++ b/models/model.py @@ -223,16 +223,16 @@ class Model: if self.image_log_on: i_a, i_c, i_p = images self.writer.add_images( + 'Appearance image', i_a, self.curr_iter + ) + self.writer.add_images( 'Canonical image', i_c, self.curr_iter ) - for (i, (o, a, p)) in enumerate(zip(x_c1, i_a, i_p)): + for i, (o, p) in enumerate(zip(x_c1, i_p)): self.writer.add_images( f'Original image/batch {i}', o, self.curr_iter ) self.writer.add_images( - f'Appearance image/batch {i}', a, self.curr_iter - ) - self.writer.add_images( f'Pose image/batch {i}', p, self.curr_iter ) time_used = datetime.now() - start_time |