完成图像形态学操作

basic/urls.py

@@ -20,4 +20,5 @@ urlpatterns = [
     path('lcd', views.line_change_detection),
     path('smooth', views.smooth),
     path('sharpen', views.sharpen),
+    path('morphology', views.morphology)
 ]

basic/views.py

@@ -810,5 +810,28 @@ def sharpen(request):
     return HttpResponse('请使用POST方法')
 
 
+@csrf_exempt
+def morphology(request):
+    if request.method == 'POST':
+        para = json.loads(request.body)
+        image = para['img']
+        size = para.get('size', 5)
+        img = cv2.imread(PREFIX + image, cv2.IMREAD_UNCHANGED)
+        kernel = cv2.getStructuringElement(cv2.MORPH_CROSS, (size, size))
+        op = para['type']
+        if op == 'open':
+            img = cv2.morphologyEx(img, cv2.MORPH_OPEN, kernel)
+        elif op == 'close':
+            img = cv2.morphologyEx(img, cv2.MORPH_CLOSE, kernel)
+        elif op == 'erode':
+            img = cv2.erode(img, kernel)
+        elif op == 'dilate':
+            img = cv2.dilate(img, kernel)
+        filename = getImageName() + DEFAULT_FORMAT
+        cv2.imwrite(PREFIX + filename, img)
+        return HttpResponse(filename)
+    return HttpResponse('请使用POST方法')
+
+
 def r(request):
     return render(request, 'upload.html')