search

songdejia / Siamese-RPN-pytorch

This is a re-implementation of Siamese-RPN with pytorch, which is CVPR2018 spotlight.
MIT License
379 stars 87 forks source link

about training and test #1

Closed vincennnnt closed 5 years ago

vincennnnt commented 5 years ago

Sorry to bother u,but i have some questions about the implementation of SiamRPN.When i use zkisthebest's code,the performance is so bad ,so i tried to add ur code into his,but when i train the code,it alerts the error TypeError: batch must contain tensors, numbers, dicts or lists; found <class 'PIL.Image.Image'>.So i wonder whether u can provide train.py and test.py

songdejia commented 5 years ago

Thanks for your attention. Now this work is not yet complete, some parts need to be modified and tested. I will update later, please pay attention to this work.

vincennnnt commented 5 years ago

@songdejia Thank u so much！！！

songdejia commented 5 years ago

We have just finished v1.0 on pytorch and tensorflow.
pytorch-version is available here [https://github.com/songdejia/siamese-RPN] tensorflow-version is available here [https://github.com/makalo/Siamese-RPN-tensorflow.git] @vincennnnt

vincennnnt commented 5 years ago

@

vincennnnt commented 5 years ago

@songdejia Thank u for ur great job！！