JJGO/UniverSeg

Bug in visualize_tensors function

Closed this issue · 3 comments

What is E in visualize_tensors function?

ax.imshow(E.rearrange(x,'C H W -> H W C'))

adalca commented

We had a line

import einops as E

a bit lower down. We moved it higher up to avoid the confusion.

E refers to einops

JJGO commented

Does this resolve your issue @kennyha85

Does this resolve your issue @kennyha85

Yes, the issue has been solved