import matplotlib.pyplot as plt
from Tank import Tank
import picamera
import picamera.array
import time
import pygame
from scipy import ndimage
from time import sleep
# Settings
WIDTH = 320
HEIGHT = 240
# Initialize Tank
robot = Tank()
robot.correctDirections(True,True,True)
# Initialize Pygame
pygame.init()
pygame.display.set_caption('My Robot')
screen = pygame.display.set_mode((WIDTH,HEIGHT),0)
# Image analysis
lower = np.array([25,35,70])
upper = np.array([75,255,205])
def blockAnalyze(mask):
# Assume 320,240 image
mask = np.transpose(mask)
sum = 0
count = 0
