yarmeli / finalyearproject

This contains all the relevant files for my final year university project

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

Requirements

This project requires Anaconda3 and the following libraries to run:

OpenCV2
PyTorch
TorchVision
GitLFS

Usage

1. (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

2. Clone the repo:

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

3. Run the main application:

   cd FinalYearProject
   python main.py