mobilevit-pytorch
mobilevit-pytorch copied to clipboard
Error in sequence of steps in the model
@chinhsuanwu , thank you so much for uploading the code.
-
There are small issues regarding the structure of model that you've designed, I have updated the code accordingly in my repository over here.
-
Basically, I referred to the original code present over here . I inferenced the original detector model on some sample data and compared their outputs with a detector designed from your code, at every stage. Then I found that there was a huge disparity between both of these models.
-
I have changed your code at some parts, and the results that it gave were similar to the original one, the link to the model file is here.
-
Please let me know if you want me to elaborate further on this topic.
-
The complete repository can be found here