except: print("write error.") # # main # br.show_logo() br.exit_check() br.initialize_camera() feature_file = "/sd/features.csv" feature_list,feature_0,feature_100 = load(feature_file) task = kpu.load("/sd/model/mbnet751_feature.kmodel") print('[info]: Started.') info=kpu.netinfo(task) #a=kpu.set_layers(task,29) old_name='' marker_0_100=0 clock = time.clock() try: while(True): img = sensor.snapshot() # QR Code check res = img.find_qrcodes() if len(res) > 0: name = res[0].payload() if name=="*reset":
sensor.set_framesize(sensor.QVGA) sensor.skip_frames(time=2000) # Wait for settings take effect. sensor.set_windowing((224, 224)) sensor.run(1) classes = ["person"] tinyYolo = "/sd/models/tinyYoloMerged.kmodel" #16 fps mbnet75 = "/sd/models/mbnet75Merged.kmodel" #12 fps mbnet50 = "/sd/models/mbnet50Merged.kmodel" #15fps mbnet25 = "/sd/models/mbnet25Merged.kmodel" #16fps task = kpu.load(mbnet50) try: kpu.netinfo(task) except: print("cannot get net info, kmodel v4 not supported for netinfo!") lastLayerShape = (len(classes) + 5) * 5 #from documentation Kmodel V4 need set output shape manually a = kpu.set_outputs(task, 0, 7, 7, lastLayerShape) anchor = (0.57273, 0.677385, 1.87446, 2.06253, 3.33843, 5.47434, 7.88282, 3.52778, 9.77052, 9.16828) #yolo v2 tiny def doOverlap(boxDict, box1Dict): box = boxDict['rect'] box1 = box1Dict['rect'] print("checking overlapping boxes:", box, " ", box1) box_x = box[0]