Using Machine Learning to estimate calories

This project uses Machine Learning to estimate the number of calories in a food item.

Files

File	Description
main.py	Contains the main application and the calorie estimation component
volume.py	Contains the volume estimation component and all functions related to it
model.py	Contains the image classification model and all functions related to it
image_segmentation.py	Contains the image segmentation model and all functions related to it
helpers.py	Contains helper functions - Change `DebugMode` to `False` to disable debug output
CameraWidget.py	QWidget that deals with the user's webcam information
ThumbWidget.py	QWidget that allows the user to change thumb measurements

This project requires Anaconda3 and the following libraries to run:

OpenCV2
PyTorch
TorchVision
GitLFS

(Windows) Open Anaconda Prompt and install the libraries

conda install -c menpo opencv
conda install -c pytorch pytorch torchvision
conda install -c conda-forge git-lfs

Clone the repo:

git clone https://github.com/Yarmeli/FinalYearProject.git

Name		Name	Last commit message	Last commit date
Latest commit History 104 Commits
Dataset		Dataset
Pictures		Pictures
Plots		Plots
.gitattributes		.gitattributes
.gitignore		.gitignore
CameraWidget.py		CameraWidget.py
Data Preparation.ipynb		Data Preparation.ipynb
FoodDataset Class interacting with the training data.ipynb		FoodDataset Class interacting with the training data.ipynb
Image Segmentation.ipynb		Image Segmentation.ipynb
MainWindow.ui		MainWindow.ui
README.md		README.md
ThumbWidget.py		ThumbWidget.py
Train CNN.ipynb		Train CNN.ipynb
Volume Calculations.ipynb		Volume Calculations.ipynb
helpers.py		helpers.py
image_segmentation.py		image_segmentation.py
main.py		main.py
model.py		model.py
thumb.ui		thumb.ui
thumbValues.csv		thumbValues.csv
volume.py		volume.py