2023-01-02T09:40:59.671Z

0

답변 감사드립니다. 한번 시도해 보겠습니다!

0

import cv2
import mrcnn.model as modellib
from mrcnn.visualize import InferenceConfig, get_mask_contours, random_colors, draw_mask
import numpy
import json

# Load MaskRCNN model
num_classes = 4
config = InferenceConfig(num_classes=num_classes, image_size=1024) # class 숫자와 image 사이즈 입력
model = modellib.MaskRCNN(mode="inference", config=config, model_dir="")
model.load_weights(filepath="dnn_model/mask_rcnn_object_0003.h5", by_name=True)

# Load image
img = cv2.imread("images/val06.jpg")
image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

# Detect objects
results = model.detect([image])[0]  # 우린 4개 검출해야함
class_ids = results["class_ids"]
object_count = len(class_ids)

# Random colors
colors = random_colors(object_count)

for i in range(object_count):
    # 1.class ID
    class_id = results["class_ids"][i]

    # 2.Box
    box = results["rois"][i]
    y1, x1, y2, x2 = box
    cv2.rectangle(img, (x1, y1), (x2, y2), colors[i], 2)

    # 3.Score
    score = scores = results["scores"][i]

    # 4.Mask
    mask = results["masks"][:,:,i]
    contours = get_mask_contours(mask)
    print("Contours", contours)
    for cnt in contours:
        cv2.polylines(img, [cnt], True, colors[i],2)
        img = draw_mask(img, [cnt], colors[i], alpha=0.5)

    # json dictionary
    json_dict = {
        "image_filename": img,
        "bbox": [box],
        "segmentation": [contours],
        "classes": [class_id]
    }
    json_data = json.dumps(json_dict, indent=2)
    print(json_data)

# Display image
cv2.imshow("Img", img)
cv2. waitKey(0)

이렇게 코드를 수정해서 돌려보는데

Traceback (most recent call last):

File "C:\Users\Jae\PycharmProjects\MaskRCNN_Jae\demo.py", line 52, in <module>

json_data = json.dumps(json_dict, indent=2)

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\__init__.py", line 234, in dumps

return cls(

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\encoder.py", line 201, in encode

chunks = list(chunks)

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\encoder.py", line 431, in _iterencode

yield from iterencodedict(o, currentindent_level)

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\encoder.py", line 405, in iterencodedict

yield from chunks

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\encoder.py", line 438, in _iterencode

o = _default(o)

File "C:\Users\Jae\AppData\Local\Programs\Python\Python39\lib\json\encoder.py", line 179, in default

raise TypeError(f'Object of type {o.__class__.__name__} '

TypeError: Object of type ndarray is not JSON serializable

Process finished with exit code 1

이런 에러가 나옵니다. numpy array 를 리스트로 변환해야 한다는데... tolist 를 써도 변환이 안되네요.. 혹시 한번만 살펴봐 주실 수 있나요...감사합니다.

0

image_filename 은 이미지 파일명 같은데 img를 할당하면 img는 image array로 원하는 결과가 나오지 않을 것 같습니다. image_filename을 문자열로 변경해 보시지요.