search

luuuyi / RefineDet.PyTorch

A higher performance PyTorch implementation of Single-Shot Refinement Neural Network for Object Detection
MIT License
235 stars 64 forks source link

test images meet many problems #16

Closed lixiandea closed 5 years ago

lixiandea commented 5 years ago

is anyone meet some problem about training your model with your own data? when i train the model, nothing goes wrong, but when i use my model to detect objects, i got an answer with all zeros,can and nothing goes wrong but i cant test my model on cuda()

luuuyi commented 5 years ago

@lixiandea maybe somewhere is wrong, you can check your code. If you use the demo.py, that might be wrong because the demo.py is origin version of ssd.pytorch. You should change it yourself, it's very easy.