Python MQTT-клиент для получения данных и обнаружения Pytorch для передачи результата - PullRequest
Новая
       74

Python MQTT-клиент для получения данных и обнаружения Pytorch для передачи результата

0 голосов
/ 18 февраля 2020

Недавно я создал платформу IoT, которая использует mqtt для передачи данных с датчика, и я пишу сценарий python для декодирования пакета, а затем сохраняю изображение в локальном каталоге. Кроме того, я пишу другую программу Pytorch Yolo python для обнаружения изображения. Программа приема использует функцию обратного вызова mqtt on_message для декодирования пакета, а программа обнаружения pytorch является в то же время истинной программой. Проснувшись, он обнаруживает папку, если существуют изображения или нет, и публикует sh результат в моем веб-приложении.

Мой вопрос: есть ли способ объединить эти две программы в одну, используя несколько Потоковая или многопроцессорная библиотека ??

ниже - мой код 1011 *

while True:
    while True:
        if glob.glob('../data/' + "*.jpg", recursive=False)==[]:
            time.sleep(30)
        else:
            break

    try:

        with torch.no_grad():
            bbox_list = detect(cfg=cfg_path,\
                                weights=weights_path,\
                                output=output_path,\
                                data=data_path,\
                                source=source_path,\
                                conf_thres= 0.3,\
                                nms_thres=0.5,\
                                img_size=416,\
                                half=False,\
                                fourcc='mp4v',\
                                device='',\
                                view_img=False,\
                                save_txt=False,\
                                save_img=True)

        ...
        mqtt.publish("topic/xxx",result)

Пожалуйста, дайте мне совет и помощь ~~

...