cs401r_w2016:lab8

To understand particle filters, and to gain experience with debugging likelihoods.

You may wish to install the python `pyqtgraph`

package. This can be installed via `conda install pyqtgraph`

. This is a plotting package that is significantly faster than `matplotlib`

when scatter plotting many points (ie, particles).

For this lab, you will implement a particle filter for robot localization. Your notebook should print a visualization of the true location of the robot, your estimate of its position, and a simple scatterplot of the distribution of particles at the end of tracking. For example, my visualization looks like the following:

To successfully complete the lab, you will have to:

- Create an initial set of particles
- Create a proposal distribution
- Create an observation likelihood distribution
- Given a set of particles, compute the expected value of the x,y position of the robot
- Resample the particles to avoid weight degenercy

Your notebook will be graded on the following elements:

- 25% Implemented a correct sensor likelihood
- 25% Implemented a reasonable proposal distribution
- 30% Implemented particle resampling correctly
- 10% Correctly compute expected value of state x,y coordinates
- 10% Final plot is tidy and legible

For this lab, you will implement the basic particle filter found in Algorithm 23.1 of the book (pg. 826). You can assume that your particle proposal distribution is equal to the dynamics prior (see Section 23.5.4), which leads to a very simple form of particle filtering. Specifically, you may assume that the “dynamics” are that the moves in a random direction and changes angles randomly. I used $p(z_{t+1}|z_{t}) = \mathcal{N}(z_{t},\Sigma)$, where $\Sigma$ is a parameter you will need to tune.

You will need the following files:

sensors.mat contains the data you will use (you can also regenerate it directly, using the script below). The data is in the `sonars`

field, and the true states are in the `true_states`

field. `true_states`

is a 3xT matrix, where each 3×1 column represents the true x,y and angle (\in 0,2*pi) of the robot at time t.

lab8_common.py contains utility functions that will enable you to create a simple map, and compute what a robot would see from a given position and angle.

lab8_gen_sonars.py is the script that I used to generate the data for this lab. If you run it, it will generate a new dataset, but it will also produce a nice visualization of the robot as it moves around, along with what the sensors measure.

To get started, check out the provided functions in lab8_common.py. There are two critical functions:

`create_map`

will create a map object and return it`cast_rays`

will simulate what a robot would see from a specific x,y,theta position, given a map

`cast_rays`

is fully vectorized, meaning that you can simulate from multiple locations simultaneously. For example, in my code, I only call `cast_rays`

once:

part_sensor_data = cast_rays( parts, room_map ) liks = ray_lik( data[:,t:t+1], part_sensor_data.T ).T

where I have represented my particles as a 3xN matrix. In the above example call, `part_sensor_data`

is an Nx11 matrix. `ray_lik`

is my implementation of the likelihood function $p(y_t|z_t)$, and is also fully vectorized. It returns an Nx1 matrix, with the likelihood of each particle.

Your proposal distribution can be anything you want. Be creative. You may have to experiment!

You should experiment with two different likelihoods. First, try using a simple Gaussian likelihood. You will need to tune the variance, but don't spend too long on it; I never got this to work very well. Second, try using a simple Laplacian distribution. (The Laplacian is like a Gaussian, except that instead of squaring the difference, one uses an absolute value): $$p(y_t|z_t^s) = \exp \left( -\dfrac{1}{\sigma} \sum_i \lvert (y_t)_i - \mathrm{cast\_rays}(z_t^s)_i \rvert \right)$$, where $\sigma$ is a parameter that you will need to tune.

You will have to experiment with the parameters of these distributions to make them work. Try values between 0.01 and 1.

You may find the `numpy.random.mtrand.multinomial`

, or `numpy.random.choice`

functions helpful.

It can be frustrating to experiment with a particle filter when the interface is too slow. By using `pyqtgraph`

instead of `matplotlib`

, you can speed things up considerably.

Here is some code to create a `pyqtgraph`

window, and show the map.

from pyqtgraph.Qt import QtGui, QtCore import pyqtgraph as pg app = QtGui.QApplication( [] ) win = pg.GraphicsWindow( title="Particle filter" ) win.resize( 600, 600 ) win.setWindowTitle( 'Particle filter' ) pg.setConfigOptions( antialias=True ) p3 = win.addPlot( title="Room map" ) for i in range( 0, room_map.shape[0] ): p3.plot( [room_map[i,0], room_map[i,2]], [room_map[i,1], room_map[i,3]] ) p3.setXRange( -0.1, 1.1 ) p3.setYRange( -0.1, 1.1 )

To speed things up further, you should use the `.setData`

method on your plots. This replaces the data used for a particular plotting element, and is faster than deleting and recreating the plot element. This is particularly important when trying to update many particles.

Here's a snippet of my code. `ts_plot`

and `ex_plot`

are initialized to `None`

. My particles are stored in a 3xN matrix called `parts`

, my expected states are stored in a 3xt matrix called `exs`

.

if ts_plot == None: ts_plot = p3.plot( true_states[0,0:t+1], true_states[1,0:t+1], pen=(0,0,255) ) ex_plot = p3.plot( exs[0,0:t+1], exs[1,0:t+1], pen=(0,255,0) ) pts = p3.scatterPlot( parts[0,:], parts[1,:], symbol='o', size=1, pen=(255,100,100) ) else: ts_plot.setData( true_states[0,0:t+1], true_states[1,0:t+1] ) ex_plot.setData( exs[0,0:t+1], exs[1,0:t+1] ) pts.setData( parts[0,:], parts[1,:] ) pg.QtGui.QApplication.processEvents()

(you must call `processEvents`

to actually update the display)

The combination of these two things allowed me to experiment with my particle filter in real-time. Note that you are not required to produce any sort of visualization over time, just a single plot at the end of running your code. The use of `pyqtgraph`

is strictly for the work you will do debugging your particle filter.

Or if you want to use matplotLib which is better for Jupyter Notebook but slower, here is some code for plotting at each iteration:

import matplotlib.pyplot import numpy from IPython.display import clear_output def draw(particles, true_values, predicted_values, t): clear_output(wait=True) show_map( room_map ) plt.scatter(particles[0,:] , particles[1,:], c="yellow") plt.plot(true_values[0,:t+1], true_values[1,:t+1] , c= "green", label="Actual Position") plt.plot(predicted_values[0,:t+1], predicted_values[1,:t+1], c= "red", label="Estimated Position") plt.legend(loc=1) plt.show()

cs401r_w2016/lab8.txt · Last modified: 2021/06/30 23:42 (external edit)