浏览量：2608 最近编辑于：2021-05-27 19:29:22
# FRS-pytorch
Food Recognition System using pytorch deep learning framework??????????????????
* Deep Learning based food recognition system, using `Pytorch` deep learning framework
* This project can be also transplanted to other platforms like `Raspberry Pi`
[https://github.com/ivanwhaf/FRS-pytorch](https://github.com/ivanwhaf/FRS-pytorch "https://github.com/ivanwhaf/FRS-pytorch")
# Demo
## 1. Run demo directly
*cam_demo.py* runs to show a UI and recognize the food?:
```bash
$ python cam_demo.py
```
## 2. Run in command line
*detect.py* runs to inference on a variety of sources, cd your project path and type:
```bash
$ python detect.py -c cfg/frs.cfg -w weights/frs_cnn.pth --source 0 # webcam
{filename}.jpg # image
{filename}.mp4 # video
{path}/{filename}.jpg # path of image
{path}/{filename}.mp4 # path of video
{folder} # image folder
```
# Usage
## Preparation
* 1. Create an empty folder (this demo is `dataset` folder) as your dataset folder
* 2. Prepare your own image dataset, or you can run spiders `spider_baidu.py` and `spider_douguo.py` (in utils folder) to crawl raw images from the internet
* 3. Move your raw image data into dataset folder, dataset folder contain several sub folders, each
sub folder represents each class, for example your dataset folder should be like this:
```
dataset/
├──tomato/
| ├──001.png
| ├──002.png
| └──003.jpg
└──potato/
├──001.png
├──002.png
└──003.jpg
```
## Training
### 1. Create a config file
Create an empty config file `xxx.cfg` (xxx is your project name) in `cfg` directory (this repo is *cfg/frs.cfg*), then imitate `frs.cfg` editing customized configs. Set **num_classes** according to your class number, set **dataset** as your dataset path, set **input_size** as your image input size
### 2. Choose a network model
Choose a network model in `models` folder, and edit model parameters in `frs.cfg`, for example **model: ResNet18**. You can also customize your own model and add it to `models` folder. Don't forget modifying function `build_model()` in `model.py`
### 3. Modify hyper parameters
Enter `train.py` and edit **epochs**, **learning_rate**, **batch_size**, **input_size** and other hyper parameters as you wish
### 4. Train
Run `train.py` to train your own model (only when dataset was prepared):
```bash
$ python train.py --cfg cfg/frs.cfg
```
## Caution
* Need plotting model structure? Just install `graphviz` and `torchviz` first
* Please screen out unqualified raw images manually when making dataset
* Validation and test periods are among training process, please see train.py for more details
# Program Structure Introduction
* cfg: some config files
* data: some samples and misc files
* dataset: your own dataset path
* models: some network model structures
* spiders: some python spiders for downloading images
* utils: some util and kernel files
* output: output file folder
* weights: model weights
# Requirements
Python 3.X version with all [requirements.txt](https://github.com/ivanwhaf/FRS-pytorch/blob/master/requirements.txt) dependencies installed, including `torch>=1.2`. To install run:
```bash
$ pip install -r requirements.txt
```
# Environment
## PC Ⅰ
* Windows 10
* Python 3.8.6
* CUDA 10.1
* cuDNN 7.6
* torch 1.7.0
* Nvidia GTX 1080Ti 11G
## PC Ⅱ
* Windows 10
* Python 3.6.8
* CUDA 10.2
* cuDNN 7.6
* torch 1.6.0
* Nvidia GTX 1060 3G
## PC Ⅲ
* Ubuntu 20.04
* Python 3.7.8
* CUDA 10.0
* cuDNN 7.4
* torch 1.2.0
* Nvidia MX350 2G
评论