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]