| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- #!/usr/bin/env python
- #version : 2023.12.31
- #language : en
- from time import time
- from maix import image
- class Person:
- mud_path = "/root/preset/model/person_int8.mud"
- labels = ["person"]
- anchors = [4.72, 6.26, 1.39, 3.53, 0.78, 1.9, 0.35, 0.95, 2.49, 4.87]
- def __init__(self) -> None:
- from maix import nn
- self.model = nn.load(self.mud_path)
- from maix.nn import decoder
- self.decoder = decoder.Yolo2(len(self.labels) , self.anchors , net_in_size = (224, 224) ,net_out_size = (7,7))
- def __del__(self):
- del self.model
- del self.decoder
- def cal_fps(self ,start , end):
- one_second = 1
- one_flash = end - start
- fps = one_second / one_flash
- return fps
- def draw_rectangle_with_title(self ,img, box, disp_str , fps ):
- img.draw_rectangle(box[0], box[1], box[0] + box[2], box[1] + box[3],color=(255, 0, 0), thickness=2)
- img.draw_string(box[0], box[1]+ box[3] ,disp_str, scale=1,color=(0, 0, 255), thickness=2)
- #img.draw_string(0, 0 ,'FPS :'+str(fps), scale=2 ,color=(0, 0, 255), thickness=2)
- def process(self,input):
- t = time()
- out = self.model.forward(input, quantize=1, layout = "hwc")
- boxes, probs = self.decoder.run(out, nms=0.5, threshold=0.6, img_size=(224,224))
- for i, box in enumerate(boxes):
- class_id = probs[i][0]
- prob = probs[i][1][class_id]
- disp_str = "{}:{:.2f}%".format(self.labels[class_id], prob*100)
- fps = self.cal_fps(t, time())
- self.draw_rectangle_with_title(input, box, disp_str, fps)
- canvasImg = image.new(size = (320, 240))
- def main():
- from maix import display, camera
- app = Person()
- #camera.config((224,224))
- while True:
- canvasImg.clear()
- img = camera.capture().crop(0, 0,224, 224)
- app.process(img)
- canvasImg.draw_image(img,48,8)
- canvasImg.draw_image(image.open("/root/preset/img/exit_ff0000_24x24.png"),288,216,alpha=1)
- display.show(canvasImg)
- main()
|
