首页 > 解决方案 > 虚假陈述的范围问题

问题描述

我正在尝试让我的 Raspberry Pi 成功扫描,然后按照这个存储库跟踪 RPi 相机。当我发出“rpi-deep-pantilt track Raspi”命令时,它成功地开始在静止位置检测,当它检测到物体时,它开始跟踪。我需要它进行扫描,在检测时来回平移 60 度增量,因此我将粗体线合并到 manager.py 脚本中:

import logging
from multiprocessing import Value, Process, Manager

import pantilthat as pth
import signal
import sys
**import time
import RPi.GPIO as GPIO**

**from rpi_deep_pantilt.detect.util.visualization import draw_bounding_box_on_image**
from rpi_deep_pantilt.detect.camera import run_pantilt_detect
from rpi_deep_pantilt.control.pid import PIDController

**#Specifying GPIO Pin name
GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
GPIO.setup(8,GPIO.OUT)**

logging.basicConfig()
LOGLEVEL = logging.getLogger().getEffectiveLevel()

RESOLUTION = (320, 320)

SERVO_MIN = -90
SERVO_MAX = 90

CENTER = (
    RESOLUTION[0] // 2,
    RESOLUTION[1] // 2
)

# function to handle keyboard interrupt
def signal_handler(sig, frame):
    # print a status message
    print("[INFO] You pressed `ctrl + c`! Exiting...")

    # disable the servos
    pth.servo_enable(1, False)
    pth.servo_enable(2, False)
    GPIO.output(8,GPIO.LOW)

    # exit
    sys.exit()


def in_range(val, start, end):
    # determine the input value is in the supplied range
    return (val >= start and val <= end)


def set_servos(pan, tilt):
    # signal trap to handle keyboard interrupt
    signal.signal(signal.SIGINT, signal_handler)
    **scan_on=True**
    
    while True:
        pan_angle = -1 * pan.value
        tilt_angle = tilt.value
        
        # if the pan angle is within the range, pan
        if in_range(pan_angle, SERVO_MIN, SERVO_MAX):
            pth.pan(pan_angle)
        else:
            logging.info(f'pan_angle not in range {pan_angle}')

        if in_range(tilt_angle, SERVO_MIN, SERVO_MAX):
            pth.tilt(tilt_angle)
        else:
            logging.info(f'tilt_angle not in range {tilt_angle}')
        
        draw_bounding_box_on_image

        **while scan_on == True:
            pth.servo_one(90)
            pth.servo_two(25)
            time.sleep(10)
            
            pth.servo_one(30)
            pth.servo_two(25)
            time.sleep(10)
            
            pth.servo_one(-30)
            pth.servo_two(25)
            time.sleep(10)
            
            pth.servo_one(-90)
            pth.servo_two(25)
            time.sleep(10)

            continue**
.
.

最后几行是扫描程序。它们可以工作,但是一旦检测到对象,它就不会退出扫描周期。这是导入到 manager.py 的 visualizion.py 脚本的功能:

def draw_bounding_box_on_image(image,
                               ymin,
                               xmin,
                               ymax,
                               xmax,
                               color='red',
                               thickness=4,
                               display_str_list=(),
                               use_normalized_coordinates=True):


    **GPIO.output(8,GPIO.HIGH)
    scan_on = False**

    draw = ImageDraw.Draw(image)
    im_width, im_height = image.size
    if use_normalized_coordinates:
        (left, right, top, bottom) = (xmin * im_width, xmax * im_width,
                                      ymin * im_height, ymax * im_height)
    else:
        (left, right, top, bottom) = (xmin, xmax, ymin, ymax)
    draw.line([(left, top), (left, bottom), (right, bottom),
               (right, top), (left, top)], width=thickness, fill=color)
    try:
        font = ImageFont.truetype('arial.ttf', 24)
    except IOError:
        font = ImageFont.load_default()

    # If the total height of the display strings added to the top of the bounding
    # box exceeds the top of the image, stack the strings below the bounding box
    # instead of above.
    display_str_heights = [font.getsize(ds)[1] for ds in display_str_list]
    # Each display_str has a top and bottom margin of 0.05x.
    total_display_str_height = (1 + 2 * 0.05) * sum(display_str_heights)

    if top > total_display_str_height:
        text_bottom = top
    else:
        text_bottom = bottom + total_display_str_height
    # Reverse list and print from bottom to top.
    for display_str in display_str_list[::-1]:
        text_width, text_height = font.getsize(display_str)
        margin = np.ceil(0.05 * text_height)
        draw.rectangle(
            [(left, text_bottom - text_height - 2 * margin), (left + text_width,
                                                              text_bottom)],
            fill=color)
        draw.text(
            (left + margin, text_bottom - text_height - margin),
            display_str,
            fill='black',
            font=font)
        text_bottom -= text_height - 2 * margin
    return scan_on

我使用 GPIO 点亮了一个 LED,让我知道边界框何时覆盖在相机馈送上。此时我的系统应该识别出它scan_on = False,它应该退出扫描周期并恢复到默认跟踪模式。我的 manager.py 脚本不能拾取它,因为它会继续运行扫描,这意味着scan_on = True.

标签: pythonfunctionif-statement

解决方案


推荐阅读