- Python >= 3.7
- Pytorch >= 1.12 is recommended
- opencv-python
- sklearn
- matplotlib
Use the below command to train the model on our database.
python ./main.py
Use the below command to test the model on our database.
python ./test_main.py
https://www.kaggle.com/datasets/liuhanyu1007/m3c2-data
https://www.kaggle.com/models/liuhanyu1007/m3c2_model