masonkadem / remote-pupillometry

Pupil Size Measurement using PyTorch

This repository contains code for a real-time pupil size measurement system using PyTorch and OpenCV.

Setup and Installation

1. Clone the repository:

git clone https://github.com/masonkadem/PupilSizeMeasurement.git

2. Navigate to the repository:

cd PupilSizeMeasurement

3. Install the necessary Python libraries:

pip install -r requirements.txt

Running the Application

To run the application, use the following command:

python pupil.py

The script will start the webcam and begin detecting the pupils in real-time.

Docker Instructions

You can also run the application in a Docker container.

1. Further DevelopmentBuild the Docker image:

docker build -t pupil-size-measurement .

2. Run the Docker container:

docker run --rm -p 5000:5000 --name=pupil-size-measurement pupil-size-measurement

Further Development

This application can be extended and improved in several ways, such as improving the accuracy of the pupil size measurement, optimizing the performance for real-time detection, and incorporating more advanced pupil detection methods.

remote-pupillometry's Issues

Missing files

Hi!
I would use the code for a university psychology experiment, but I couldn't test it because the requirements.txt is missing, and I also ran into some error messages. I'm not very familiar with the different packages, so I'll write down that the following error messages were thrown when running "torch":
Remote-Pupillometry-main/pupil.py", line 15, in
checkpoint = torch.load('prop_model.pth')
Python38\lib\site-packages\torch\serialization.py", line 791, in load
with _open_file_like(f, 'rb') as opened_file:
Python38\lib\site-packages\torch\serialization.py", line 271, in _open_file_like
return _open_file(name_or_buffer, mode)
Python38\lib\site-packages\torch\serialization.py', line 252, in init
super().init(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: 'prop_model.pth'

Can you help me with these, what did I do wrong? Thank you very much!
Best regards:
János Darnay
BME - Hungary