prakrutpatel commited on
Commit
e304239
·
1 Parent(s): d186fd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -399,12 +399,12 @@ def segment(image):
399
  plt.rcParams['ytick.left'] = False
400
  plt.rcParams['ytick.right'] = False
401
  plt.rcParams['figure.figsize'] = [15,10]
402
- date_captured = Image.open(image)._getexif()[36867]
403
  context_feature, score, results = run_inference(
404
  faster_rcnn_model, image, date_captured)
405
  faster_rcnn_output_dict = results
406
  context_rcnn_image, faster_rcnn_image = show_context_rcnn_inference(
407
- context_rcnn_model, image, context_features_matrix,
408
  faster_rcnn_output_dict, context_padding_size)
409
  plt.subplot(1,2,1)
410
  plt.imshow(faster_rcnn_image)
 
399
  plt.rcParams['ytick.left'] = False
400
  plt.rcParams['ytick.right'] = False
401
  plt.rcParams['figure.figsize'] = [15,10]
402
+ date_captured = Image.open(image.name)._getexif()[36867]
403
  context_feature, score, results = run_inference(
404
  faster_rcnn_model, image, date_captured)
405
  faster_rcnn_output_dict = results
406
  context_rcnn_image, faster_rcnn_image = show_context_rcnn_inference(
407
+ context_rcnn_model, image.name, context_features_matrix,
408
  faster_rcnn_output_dict, context_padding_size)
409
  plt.subplot(1,2,1)
410
  plt.imshow(faster_rcnn_image)