search

hellochick / PSPNet-tensorflow

TensorFlow-based implementation of "Pyramid Scene Parsing Network".
326 stars 123 forks source link

feed_dict img in order to sequential sess.run() with multiple images #66

Closed kiitosu closed 5 years ago

kiitosu commented 5 years ago

I'd like to sequential sess.run() with multiple images. image like below.

for file in files: 
  sess.run(feed_dict{data:img})

I think I need to change the code of inference.py line 73

net = PSPNet({'data': img}, is_training=False, num_classes=num_classes)

and in model.py some where around line 14 and add some kind of placeholder function in network.py .

But I'm a beginner of tensorflow, so I can't do that now. Please let me know if anyone knows the solution.