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