Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

snn output doesn't match the detection head #55

Open
DuanDexin opened this issue Mar 1, 2023 · 4 comments
Open

snn output doesn't match the detection head #55

DuanDexin opened this issue Mar 1, 2023 · 4 comments

Comments

@DuanDexin
Copy link

I use 'yolov3-tiny-mp2conv-mp1none-lk2relu-up2tconv' configuration to train a normal ANN model and reached a roughly 39% mAP in coco dataset, then I run 'ann_to_snn.py' script and have a transferred SNN and I got a huge .pth snn model file. Everything seems fine till now.
But when I run the 'snn_evaluate' function in 'ann_to_snn.py' script, in output_snn1, output_snn2 = snn(data) # two branches, it goes wrong. I found that the snn(data) is a 19 elements tuble and I don't know how to let these 19 output match the output_snn1 and outpu_snn2. Can you help me?
Yours faithfully

@123456ZF
Copy link

@DuanDexin It may be that the 30 series graphics card does not support ann_to_snn.py, so there will be an error, and I am experiencing the same problem.

@123456ZF
Copy link

@DuanDexin My friend ran successfully with 1650, 2060

@Bender-L
Copy link

You can try pytorch1.4 with python 3.7

@123456ZF
Copy link

@Bender-L Hello, how can I use the converted weights file of ann-snn for object detection, can it be used for detection?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants