Update utils/vision.py

utils/vision.py

@@ -49,10 +49,10 @@ def vis_face(im_array, dets, landmarks, face_size, save_name):
                 cir1 = Circle(xy=(landmarks_one[j, 0], landmarks_one[j, 1]), radius=face_size/12, alpha=0.4, color="red")
                 pylab.gca().add_patch(cir1)
 
-        #pylab.savefig(save_name)
-        #只保存图片内容，不保存坐标轴
-        pylab.axis('off')
-        pylab.savefig(save_name, bbox_inches='tight', pad_inches=0.0)
-        pylab.show()
-        # 返回图片对象
-        return pylab
+    #pylab.savefig(save_name)
+    #只保存图片内容，不保存坐标轴
+    pylab.axis('off')
+    #pylab.savefig(save_name, bbox_inches='tight', pad_inches=0.0)
+    #pylab.show()
+    # 返回图片对象
+    return pylab.gcf()