Ehsan Poormohammady
Results
3
comments of
Ehsan Poormohammady
You might add `if output.size()[1] == prediction.size()[1]:` condition, before`output = torch.cat((output,prediction))`, inside detect.py
I have this issue as well and looking for a solution but it seems that it is not supported yet :(
I used Showcaseview library which supports this: https://github.com/amlcurran/ShowcaseView