adding needed module pyplot

mmkamani7 · Feb 20, 2018 · 98e7493 · 98e7493
1 parent bb8f677
commit 98e7493
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skeletonContext.py b/skeletonContext.py
@@ -197,6 +197,7 @@ def bookstien(X, Y, beta_k=None):
 
 def SC_plot(SC, nbins_theta, nbins_r, r_inner, r_outer, N=20):
     '''Plotting polar histogram of skeleton context for each point'''
+    import matplotlib.pyplot as plt
     if len(SC) != nbins_theta * nbins_r:
         raise ValueError('dimension mismatch, check the number of bins provided')
     SC_mat = SC.reshape(nbins_theta, nbins_r)
@@ -227,4 +228,4 @@ def SC_plot(SC, nbins_theta, nbins_r, r_inner, r_outer, N=20):
     ax.set_yticks(rbins)
     ax.set_xticks(thetabins)
     plt.show()
-    return
+    return