fix interactiveseg
This commit is contained in:
@@ -341,6 +341,13 @@ def run_plugin():
|
|||||||
as_attachment=True,
|
as_attachment=True,
|
||||||
attachment_filename=form["filename"],
|
attachment_filename=form["filename"],
|
||||||
)
|
)
|
||||||
|
if name == InteractiveSeg.name:
|
||||||
|
return make_response(
|
||||||
|
send_file(
|
||||||
|
io.BytesIO(numpy_to_bytes(bgr_res, "png")),
|
||||||
|
mimetype="image/png",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
if name == RemoveBG.name:
|
if name == RemoveBG.name:
|
||||||
rgb_res = cv2.cvtColor(bgr_res, cv2.COLOR_BGRA2RGBA)
|
rgb_res = cv2.cvtColor(bgr_res, cv2.COLOR_BGRA2RGBA)
|
||||||
|
|||||||
Reference in New Issue
Block a user