search

peterlee0127 / tensorflow-nvJetson

TensorFlow for NVIDIA Jetson, also include patch and script for building.
https://tfjetson.peterlee.app
205 stars 61 forks source link

enabled package supports #14

Closed gustavz closed 6 years ago

gustavz commented 6 years ago

How are these wheel files build from source, which package supports are enabled?

peterlee0127 commented 6 years ago

I forgot the setting for the wheel... I will write notes on next release.

gustavz commented 6 years ago

alright no porblem. I got another question: Did you try to run Tensorflows Object detection or Deeplab Models with your 1.7 wheel file installed? For me it does only work with your 1.6 version and even there Deeplab does not work.

Would be great if you would try it and tell me if it works for your or not!

Here is a small project to use deeplab for either demo segmentation on test images or on video stream: https://github.com/GustavZ/realtime_segmenation

peterlee0127 commented 6 years ago

Hi, I have no idea about this. :( But in test.py 109. You should pass config parameter to tf.Session

104         # TF config
105         config = tf.ConfigProto(allow_soft_placement=True)
106         config.gpu_options.allow_growth=True
107         print("> Starting Segmentaion")
108         with detection_graph.as_default():
109                 with tf.Session(config=config,graph=detection_graph) as sess: