The objective of this project is to add an additional camera and array of ultrasonic sensors to the base model car. These additional sensors will interface with the donkeycar software and allow the car to identify and react to common city driving events. The following steps were taken to accomplish this task:
- Brushless to Brushed DC Motor drive train conversion
- Interface two Raspberry Pi 3 Model Bs
- Use OpenCV to identify and classify common road signs
- Implement all tasks into the DonkeyCar framework
William Liu - firstname.lastname@example.org
Jingpei Lu - email@example.com
Gates Zeng - firstname.lastname@example.org
The brushless DC motor (BLDC) provided on the original chassis has one fundamental problem, it cannot be controlled effectively at low speeds. This makes it difficult accurately emulate congested city driving environments. While the BLDC performed well for high speed driving, its lack of accuracy and precision was not ideal for this project. We found two solutions for this project. One, use a Sensored BLDC for improved control while maintaining top end performance. Two, use a Brushed DC Motor for improved control, but sacrifice top end performance.
The Brushed DC Motor was chosen as its top end performance was adequate for the scope of the project and, more importantly, its cost is an order of magnitude less compared to a Sensored BLDC.
Initially, we had planned to port all our electronics onto a smaller chassis and convert the Brushless DC Motor Drivetrain into a Brushed DC Motor Drivetrain, as the Brushed DC Motor was readily available. The drivetrain conversion on the small chassis proved to be a success, we were able to achieve accurate and precise control of the throttle. There was one drawback of using the smaller chassis; first, although the car is smaller, the turning radius on the 1:10 scale car was much worse compared to the 1:8 scale car, secondly, if we were to use the smaller car, we would need to retain all of our self driving models to achieve autonomous driving again. Therefore, our next step was to look at a Brushed DC Motor conversion for the 1:8 scale car.
There were three main considerations we took when we looked for a Brushed DC Motor suitable for the car.
- The motor needs to fit and mount onto the current motor mount.
- The motor needs to provide enough power to drive the car.
- The purchased motor needs to arrive in a timely matter.
Each of these constraints proved to be a challenge, as we could easily find a motor that fit two, but not three of the constraints.
For the physical dimensions of the motor, we need to find a motor that had a diameter of 35.8 ± 4.0mm dia., a length of 73.7 ± 4.0mm, and a shaft diameter of 5.0mm exactly.
The original BLDC had a Max Current of 90A, Max Voltage of 27V, Max Power of 2400W, KV of 2200 and Max RPM of 50000rpm. Since we were looking for a Brushed Motor, we looked for something with at least 10% of the original motor's power.
Unfortunately, most of the motors we found that fit all the criteria, did not meet the last: shipping time. Therefore, we looked to do make things work with what we had on hand; by converting the 1/10 scale motor to work on the 1/8 scale car.
Designing the Shaft Adapter
The 1/10 scale motor bolted directly into the 1/8 scale car's motor mounts. The only dimension that did not match was the diameter of the shaft. The 1/10 scale motor had a 3mm dia. shaft, while the 1/10 scale car required a 5mm dia. shaft.
Because the 1/10 scale motor used a round shaft, we were not sure if we could design an adapter that would not slip while driving. However, we wanted to try it anyways.
Proof of Concept
We first constructed an adapter out of paper, this provided a cheap and easy way to see if an adapter would work. The paper adapter did not slip, however, since it was made of paper, the fitment was not perfect and there was no way to provide a proper hole for a set screw, we observed a lot of eccentric rotation of the pinion while driving.
Next, we looked to design an adapter using the 3D printers. This allowed us to make a perfect fitting adapter that also includes a feature for the set screw fit and mate with the shaft. This eliminated the eccentric rotation of the pinion and produced great performance.
Interfacing the Pi's
While a Raspberry Pi is a widely available and affordable platform for an autonomous vehicle, it does have hardware limitations that required our team to use a second Pi. One Pi will be used to run the DonkeyCar software to autonomously drive the car. The second Pi will be used to process traffic signs, and read data from the ultrasonic sensors. We needed a way to interface the two Pi's so that they could effectively communicate with each other. There were many solutions to this problem, we considered communication over WiFi, Ethernet, I2C and UART. We ended up choosing UART since the implementation seemed easy and straightforward.
First, three female to female jumper cables were connected to the Pi's in the following configuration:
TX <-> RX
RX <-> TX
GND <-> GND
To test that serial communication between the Pi's was working, Minicom was installed, UART was enabled, and console service was removed on both Raspberry Pi's.
sudo apt-get install minicom
To Enable UART, open "/boot/config.txt"
sudo nano /boot/config.txt
Add the lines below at the end of the file
# Enable UART enable_uart=1
Disabling Console Service
Disable serial-getty service
sudo systemctl disable serial.getty@ttyS0.service
sudo nano /boot/cmdline.txt
Remove "console=serial0,115200", then save the file.
Testing Serial Communication
Launch minicom on both Pi's by running:
sudo minicom -b 115200 -o -D /dev/ttyS0
To see what you're typing on minicom, turn on Local Echo through the menu by typing Ctrl-A, then E.
You should now be able to type on one Pi, see the same text in the other Pi, and vice versa.
To install pyserial
sudo apt-get install python-serial
To test that Pyserial works, we made a simple program that sent strings from one Pi to another.
This program takes an input and sends it to the other Pi:
import serial import time import io ser = serial.Serial( port = '/dev/serial0', \ baudrate = 115200, \ bytesize = serial.EIGHTBITS, \ timeout = 0) while True: input = raw_input("Input: ") ser.write(input) time.sleep(0.1) ser.close()
This program listens for an input, decodes it, and prints it out to the console:
import serial import io ser = serial.Serial( port = '/dev/serial0' ,\ baudrate = 115200 ,\ bytesize = serial.EIGHTBITS ,\ timeout = 0) print("connected to: " + ser.port) temp = '' while True: for c in ser.read(1).decode('utf-8'): temp = temp + c #print(temp) if c == '\n': print(temp) temp = '' ser.close()
Detecting Road Signs
To detect the traffic signs, we are using the Haar cascade classifier in OpenCV. Haar feature-based cascade classifiers is a machine learning based approach and it is achieved by Adaboost. OpenCV comes with a trainer as well as a detector of cascade classifier, so it is easy to train and use. Before working on the cascade classifier, you have to setup OpenCV in Python environment. You need to install OpenCV and Numpy on your computer. Then you have to clone the latest version of OpenCV in your working directory.
git clone https://github.com/Itseez/opencv.git
Training Haar Cascade Classifier
To train a Haar cascade classifier we first have to prepare the training data. The train data includes positive samples and negative samples. The positive samples are the images containing the actual objects we want to detect. The easiest way to create positive images is using the opencv_createsamples application. You can find the details about this application here. The negative samples are arbitrary images, not containing objects we want to detect. You can create these images manually or extract from videos.
When we have those samples, we have to create the description files for both positive and negative samples. The format and details of the description files can be found here. After we have the positive samples, negative samples, and the description files, we can start to train our model. For training our model, we will use the opencv_traincascade application. The details of how to use this command to train the classifier can be found here.
For details and questions about how to train a Haar cascade classifier, check the tutorial below:
Detecting Traffic Signs
After the training, we will have .xml files which are the models for the classifier. To use these models, we have to copy the .xml files to our project folder. Then, we can start a python script to detect the traffic signs.
First, we have to import the OpenCV library and Numpy library.
import numpy as np import cv2
Then, we can load our classifier. For example, if we want to load the classifier of the stop sign which is named stopsign_classifier.xml, we can do this:
stop_detect = cv2.CascadeClassifier('./stopsign_classifier.xml')
We want to detect the stop signs in videos. In the code below, we capture the video through the camera and read the images as img
cam = cv2.VideoCapture(0) ret, img =cam.read()
In this case, we are using the inbuilt webcam of the laptop. So, the ID for VideoCapture is 0. Before using the classifier, we need to convert the image to gray scale.
Now we can apply our classifier to detect the stop signs in the image
stops = stop_detect.detectMultiScale(gray,1.2,5)
Next, to visialize the detection, we can draw a rectangle on the detected stop signs and put "stop" on it.
font = cv2.FONT_HERSHEY_SIMPLEX for (x,y,h,w) in signs: cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255),2) cv2.putText(img,"stop", (x,y+h),font, 1,(255,0,0),2)
For now, the detection step is almost done. To make the classifier work in real time, we need to put the code in a loop, so the complete code is like below.
import cv2 import numpy as np stop_detect = cv2.CascadeClassifier('./stopsign_classifier.xml') font = cv2.FONT_HERSHEY_SIMPLEX cam = cv2.VideoCapture(0) while True: ret, img =cam.read() gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) stops = stop_detect.detectMultiScale(gray,1.2,5) for (x,y,h,w) in stops: cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255),2) cv2.putText(img,"stop", (x,y+h),font, 1,(255,0,0),2) cv2.imshow('im',img) if cv2.waitKey(1) & 0xFF == ord('q'): cv2.destroyAllWindows() break cam.release() cv2.destroyAllWindows()
Runing the Classifier on Raspberry Pi
To run the classifier on Raspberry Pi, we can simply import picamera and capture the image by running this:
with picamera.PiCamera() as camera: camera.resolution = (320, 240) camera.framerate = 24 time.sleep(2) image = np.empty((240 * 320 * 3,), dtype=np.uint8) camera.capture(image, 'bgr') image = image.reshape((240, 320, 3))
However, running video processing is very slow on Raspberry Pi due to the constrained computing resource. Therefore, we are using the multi-threaded approach to increase the frame rate of the video stream. In this way, we can speed up the video stream by reducing I/O latency and ensuring the main thread is never blocked.
First, we have to have a script to handle reading threaded frames from the Raspberry Pi camera module. The script can be found here. So, instead of using the picamera library, we can import the script which is named PiVideoStream.
from imutils.video.pivideostream import PiVideoStream
Then we can capture the image using the code below.
vs = PiVideoStream().start() img = vs.read()
Moreover, we know the traffic sign always appears on the right side of the vehicle so we can crop the image to half. We only run the classifier on the right half of the image.
img = img[:, (len(img)//2):]
By doing these, we can increase the FPS of our video processing for about three times.
Modifying the DonkeyCar Framework
Brief Code Overview
We start the DonkeyCar framework by calling:
python manage.py drive --model=<modelname>
Inside of manage.py, we create a Vehicle instance, which then initializes each part. One of which is the controller.
Within controller.py, there is base class named JoystickController. This provides a layer of abstraction for the different devices that are compatible with the DonkeyCar framework. The child classes will simply provide a mapping of inputs on the controller to functions and variables in the JoystickController class. In our case, PS3JoystickController is instantiated.
Inside of the JoystickController class, there is also an update() function that is polled by the ... instance. This function checks whether a button or trigger has been activated and calls the corresponding function to set the throttle, steering angle, recording setting, etc. For example, when we press 'X' on the controller, update() checks that the 'X' button was activated and then calls the emergency_stop() function which then stops recording, switches the car to user mode and stops the throttle.
Since we want to control the speed of the car, we had to find a place to plug into the code. Therefore, we added to the update() function inside of the JoystickController class. Inside update(), we read in the serial input and activate the corresponding function.
We defined several new functions for the car.
To slow the car when it sees a yield sign:
def slow_on_yield(self): print("Slowing on yield sign for 3 seconds") if (self.isAdjusted == False): self.prev_throttle = self.throttle self.throttle = self.THROTTLE_YIELD self.startTime = time.time() self.isAdjusted = True
To adjust the speed on a speed limit sign:
def adjust_on_speed(self, speed): print("Adjusting speed on speed limit sign") self.throttle = speed
To slow down the car before a stop sign:
def slow_on_stop(self): if (self.isAdjusted == False): self.prev_throttle = self.throttle self.throttle = self.THROTTLE_SLOW_STOP self.isAdjusted = True
To stop a car at the stop sign:
def stop_on_sign(self): print("Stopping on stop sign for 3 seconds") if (self.isAdjusted == False): self.prev_throttle = self.throttle self.throttle = 0.0 self.startTime = time.time() self.isAdjusted = True
Maintains the currently set throttle for the duration of 3 seconds:
def maintain_speed(self): if (time.time() - self.startTime > 3): self.isAdjusted = False self.throttle = self.prev_throttle print("Restoring speed")
In order to call these functions, we added to update() to read serial inputs and call the corresponding function:
for c in self.ser.read().decode(): # Stop sign if c == 'SIGNAL_STOP': self.stop_on_sign() # Speed limit sign elif c == 'SIGNAL_SPEED': self.adjust_on_speed(self.THROTTLE_SPEED_SIGN) # Yield sign elif c == 'SIGNAL_YIELD': self.slow_on_yield() # Slow before stop sign elif c == 'SIGNAL_PRESTOP': self.slow_on_stop() if (self.isAdjusted): self.maintain_speed()