Refer to Report.pdf for detailed explanation of project and its implementation.
- Copy the testing image files to the Folder TestImage.
- Add the pretrained yolo model(darknet) to the directory.
- Run the file test.py
python test.py - The final result i.e the prediction probability of the people using phones in the given image are stored in the FinalOutput.txt file. If it is empty then it does not have a pedestrian using phone.
- The detected humans are in the OutputImage folder and the hands cropped from the correctly classified images(clustering) are in the CropHands folder.
- The pretrained model is used in this step. The model can be retrained using the training data in JPEGImages folder. Some examples are shown in the directory.
Team Members
- Pratham Gupta
- Rajat Sharma
References for code: