wangyi1177 comments

Repositories
Issues
Comments

Results 2 comments of


                                            wangyi1177

infer() failed to predict any bbox after the first call

The infer works correctly when running tensorflow on CPU. Pretty sure something wrong when creating the input tensor on GPU : batch = tf.constant(images_data), but not sure why.

infer() failed to predict any bbox after the first call

Add os.environ["CUDA_VISIBLE_DEVICES"] = '0' or any other gpu id before creating InteractiveSession() can solve the problem. Still, not sure why. It's weird that even set os.environ["CUDA_VISIBLE_DEVICES"] = '', the session...