(logs, keys=["sample"])
| 34 | |
| 35 | |
| 36 | def logs2pil(logs, keys=["sample"]): |
| 37 | imgs = dict() |
| 38 | for k in logs: |
| 39 | try: |
| 40 | if len(logs[k].shape) == 4: |
| 41 | img = custom_to_pil(logs[k][0, ...]) |
| 42 | elif len(logs[k].shape) == 3: |
| 43 | img = custom_to_pil(logs[k]) |
| 44 | else: |
| 45 | print(f"Unknown format for key {k}. ") |
| 46 | img = None |
| 47 | except: |
| 48 | img = None |
| 49 | imgs[k] = img |
| 50 | return imgs |
| 51 | |
| 52 | |
| 53 | @torch.no_grad() |
nothing calls this directly
no test coverage detected