Problem loading color detected by arduino

0

I have a problem loading information into Arduino via pyserial.

I'm using a python and opencv script to do color detection via webcam. For each detected color, I want to inform the Arduino so he can make a decision. But I can not send the data to the Arduino.

This is the program I'm using in python

#importando módulos

import cv2   
import numpy as np
import serial
import time
import os    

#COM 4 = porta serial que está ligada o arduino
#115200 = baudrate
ser = serial.Serial('COM3', 115200, timeout = .5)

#capturando vídeo através da webcam

cap = cv2.VideoCapture(1)

while(1):
    _, imagem = cap.read()
    #quadro de conversão (imagem, ou seja, BGR) para HSV (valor de saturação de matiz)
    hsv = cv2.cvtColor(imagem, cv2.COLOR_BGR2HSV)

    #define a gama de cor vermelha
    red_lower = np.array([136, 87, 111], np.uint8)
    red_upper = np.array([180, 255, 255], np.uint8)

    #define a gama de cor azul
    blue_lower = np.array([99, 115, 150], np.uint8)
    blue_upper = np.array([110, 255, 255], np.uint8)

    #define a gama de cor verde
    green_lower = np.array([22, 60, 200], np.uint8)
    green_upper = np.array([60, 255, 255], np.uint8)

    #encontrar o intervalo de cor vermelha, azul e amarela na imagem
    red = cv2.inRange(hsv, red_lower, red_upper)
    blue = cv2.inRange(hsv, blue_lower, blue_upper)
    green = cv2.inRange(hsv, green_lower, green_upper)

    #Transformação Morfológica, Dilatação   
    kernal = np.ones((5 ,5), "uint8")
    red = cv2.dilate(red, kernal)
    res = cv2.bitwise_and(imagem, imagem, mask = red)
    blue = cv2.dilate(blue, kernal)
    res1 = cv2.bitwise_and(imagem, imagem, mask = blue)
    green = cv2.dilate(green, kernal)
    res2 = cv2.bitwise_and(imagem, imagem, mask = green)

    #Seguindo a cor vermelha
    (_, contours, hierarchy) = cv2.findContours(red, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

    for pic, contour in enumerate(contours):
        area = cv2.contourArea(contour)
        if (area > 300):
            x, y, w, h = cv2.boundingRect(contour) 
            imagem = cv2.rectangle(imagem,(x, y), (x+w, y+h), (0, 0, 255), 2)
            cv2.putText(imagem, "Vermelho",(x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255))
        if (ser.inWaiting() > 0):
            ser.write(b"2")

    #Seguindo a cor azul
    (_, contours, hierarchy) = cv2.findContours(blue, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    for pic, contour in enumerate(contours):
        area = cv2.contourArea(contour)
        if (area > 300):
           x, y, w, h = cv2.boundingRect(contour) 
           imagem = cv2.rectangle(imagem, (x, y), (x+w, y+h), (255, 0, 0), 2)
           cv2.putText(imagem, "Azul", (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 0, 0))
       if (ser.inWaiting() > 0):
          ser.write(b'1')
   #Seguindo a cor verde
   (_, contours, hierarchy) = cv2.findContours(green, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
   for pic, contour in enumerate(contours):
       area = cv2.contourArea(contour)
       if (area > 300):
           x, y, w, h = cv2.boundingRect(contour) 
           imagem = cv2.rectangle(imagem, (x, y), (x+w, y+h), (0, 255, 0), 2)
           cv2.putText(imagem, "Verde", (x, y), cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0, 255, 0))
       if (ser.inWaiting() > 0):
           ser.write(b'3')

    cv2.imshow("Seguindo a Cor", imagem)

    if cv2.waitKey(10) & 0xFF == ord('q'):
        ser.close()
        cap.release()            
        cv2.destroyAllWindows()
        break

And here's the Arduino script:

char incoming;
const int led1 = 2;
const int led2 = 4;
const int led3 = 6;

void setup() {
  Serial.begin(9600);

  pinMode(led1, OUTPUT);
  pinMode(led2, OUTPUT);
  pinMode(led3, OUTPUT);

  digitalWrite(led1, LOW);
  digitalWrite(led2, LOW);
  digitalWrite(led3, LOW);
}

void loop() {
  if (Serial.available() > 0) {
    char incoming = Serial.read();
    if (incoming == '1') {
      digitalWrite(led1, HIGH);
      Serial.println("Led 1 on");
    }
    else if (incoming == '2') {
      digitalWrite(led2, HIGH);
      Serial.println("Led 2 on");
    }
    else if (incoming == '3') {
      digitalWrite(led3, HIGH);
      Serial.println("Led 3 on");
    }
    else {}
  }
}

However, when running the program in python, the arduino only restarts and does not receive data via serial. Could someone help me with this problem?

    
asked by anonymous 17.09.2018 / 14:35

0 answers