search

see-- / keras-centernet

A Keras implementation of CenterNet with pre-trained model (unofficial)
MIT License
337 stars 84 forks source link

0it [00:00, ?it/s] no output #8

Closed caikw0602 closed 5 years ago

caikw0602 commented 5 years ago

There is no output and display "0it [00:00, ?it/s]" when I run the file "ctdet_image.py" with Pycharm. How can i figure it out please? Would you please give me some help please? Thank you very much.

see-- commented 5 years ago

I don't use Pycharm and thus can't help you with it :). The script didn't find the image(s) ->0it. Make sure you run it from the repo root dir (i.e. use keras_centernet/bin/ctdet_image.py) and that the image (s) exist.

caikw0602 commented 5 years ago

Thank you very much for your reply.Further, I also tried running the net without pycharm, unluckily OOM error occurred. My GPU is Geforce 970M with 3G memory. Is it because that the hourglass net is too big, my GPU can not afford it?

see-- commented 5 years ago

I works with 4GB memory. Try --inres 384,384 to reduce the memory usage. Good luck!

caikw0602 commented 5 years ago

Yes, I also tried that size. The OOM error still exists. Thank you for your kindly reply.

azhaoaigaga commented 5 years ago

@see-- Hello, I also use a 4gb gpu, but it is over to always create the tensorflow device, why?