README.md 1.11 KB
Newer Older
Noric Couderc's avatar
Noric Couderc committed
1
2
3
4
5
6
7
8
# Autonomous systems 1: Object detection

## The goal of the project

We want to take the [MobileNet](https://pytorch.org/hub/pytorch_vision_mobilenet_v2/) implementation with PyTorch and try to extend it.

The goal is to use some data from the [Food 101 dataset](https://www.kaggle.com/kmader/food41) and try to use it to extend MobileNet so that it can classify pictures of food as well.

Noric Couderc's avatar
Noric Couderc committed
9
## How to set up an environment with Python
Noric Couderc's avatar
Noric Couderc committed
10
11
12
13
14
15
16
17
18
19
20
21

Make sure you are using Python 3: `python --version`. On Ubuntu, you can probably use the `python3` command to resolve any ambiguities.

Create an environment directory, the packages you install with `pip` (for instance, PyTorch) will be stored there. I put it in the "env" directory.

	virtualenv -p python3 env

Activate the virtual environment:

	# If you are using bash
	source env/bin/activate

Noric Couderc's avatar
Noric Couderc committed
22
23
Your prompt should have if the activation of the virtual env worked.

Noric Couderc's avatar
Noric Couderc committed
24
25
26
## Installing PyTorch

You need to go on [the PyTorch page](https://pytorch.org/get-started/locally/#mac-anaconda) and choose from a selector the things you have, it will generate a command to paste in your terminal to install it.