|
# Acne Detection Model |
|
|
|
 |
|
|
|
## Overview |
|
|
|
This repository contains a YOLOv8-based model trained for detecting acne on African and dark skin tones. The model is designed to be inclusive, focusing on diverse datasets to improve dermatological applications' accuracy across various skin types. |
|
|
|
## Model Details |
|
|
|
- **Model Type:** YOLOv8 |
|
- **Architecture:** YOLOv8 |
|
- **Version:** 1.0 |
|
|
|
## Usage |
|
|
|
### Installation |
|
|
|
```bash |
|
pip install -r requirements.txt |
|
``` |
|
|
|
### Model Loading |
|
|
|
```python |
|
from ultralytics import YOLO |
|
|
|
# Load the model |
|
model = YOLO("acne.pt") |
|
|
|
# Perform inference |
|
result = model.detect_acne(image_path="path/to/test_image.jpg") |
|
print(result) |
|
|
|
``` |
|
|
|
Replace `"acne.pt"` with the correct model weights file. |
|
|
|
## License |
|
|
|
This project is licensed under the Apache License 2.0 - see the [LICENSE](LICENSE) file for details. |
|
|
|
## Acknowledgments |
|
|
|
- Data labeling by [Amina Shiga](https://www.linkedin.com/in/amina-shiga-07000522a) |
|
- Model Training by [Nathaniel Handan](https://www.linkedin.com/in/nathanielhandan/) |
|
|
|
## Contributing |
|
|
|
We welcome contributions from the community. If you find any issues or have suggestions, please open an [issue](https://github.com/Tinny-Robot/acne-detection/issues) or submit a pull request. |
|
|
|
## Contact |
|
|
|
For inquiries, please contact [Nathanil Handan](mailto:[email protected]). |
|
|
|
## References |
|
|
|
- Ultralytics YOLOv8: https://github.com/ultralytics/ultralytics |
|
- VGG Image Annotator (VIA): https://www.robots.ox.ac.uk/~vgg/software/via/ |