Merge branch 'master' of github.com:RentadroneCL/model-definition
This commit is contained in:
38
README.md
38
README.md
@@ -6,8 +6,8 @@
|
||||
## To do list:
|
||||
- [x] Import model detection (SSD & YOLO3)
|
||||
- [x] Model Panel Detection
|
||||
- [ ] Model Soiling Fault Detection
|
||||
- [ ] Model Diode Fault Detection
|
||||
- [x] Model Soiling Fault Detection
|
||||
- [x] Model Diode Fault Detection
|
||||
- [ ] Model Other Fault Detection
|
||||
|
||||
|
||||
@@ -20,7 +20,10 @@
|
||||
* OpenCV
|
||||
* Beautiful Soup 4.x
|
||||
|
||||
## Detection
|
||||
## Model Detection
|
||||
The models used for detection are SSD [SSD: Single Shot MultiBox Detector](https://arxiv.org/abs/1512.02325) and YOLOv3 [YOLOv3: An Incremental Improvement] (https://arxiv.org/abs/1804.02767), they are imported from the following repositories:
|
||||
* [SSD_Keras](https://github.com/pierluigiferrari/ssd_keras#how-to-fine-tune-one-of-the-trained-models-on-your-own-dataset)
|
||||
* [YOLOv3_Keras](https://github.com/experiencor/keras-yolo3)
|
||||
|
||||
Grab the pretrained weights of SSD and YOLO3 from https://drive.google.com/drive/folders/1LSc9FkAwJrAAT8pAUWz8aax_biFAMMXS?usp=sharing
|
||||
|
||||
@@ -187,8 +190,35 @@ On folder Result_ssd300_fault_1 show code (jupyter notebook), weight and result
|
||||
|
||||
|
||||
### YOLO3
|
||||
On folder Result_ssd300_fault_1 show history train (yolo3_full_yolo.output), weight and result of this model (mAP 73.02%).
|
||||
On folder Result_yolo3_fault_1 show history train (yolo3_full_yolo.output), weight and result of this model (mAP 73.02%).
|
||||
|
||||
|
||||

|
||||
|
||||
|
||||
## Diode Fault Detector
|
||||
### YOLO3
|
||||
On folder Result_yolo3_fault_4 show history train (yolo3_full_yolo.output), weight and result of this model (mAP 73.02%).
|
||||
|
||||

|
||||
|
||||
|
||||
|
||||
# Contributing
|
||||
|
||||
Contributions are welcome and will be fully credited. We accept contributions via Pull Requests on GitHub.
|
||||
|
||||
## Pull Request Checklist
|
||||
|
||||
Before sending your pull requests, make sure you followed this list.
|
||||
|
||||
- Read [contributing guidelines](CONTRIBUTING.md).
|
||||
- Read [Code of Conduct](CODE_OF_CONDUCT.md).
|
||||
- Check if my changes are consistent with the [guidelines](https://github.com/RentadroneCL/model-definition/blob/master/CONTRIBUTING.md#general-guidelines-and-philosophy-for-contribution).
|
||||
- Changes are consistent with the [Coding Style](https://github.com/RentadroneCL/model-definition/blob/master/CONTRIBUTING.md#c-coding-style).
|
||||
- Run [Unit Tests](https://github.com/RentadroneCL/model-definition/CONTRIBUTING.md#running-unit-tests).
|
||||
|
||||
|
||||
# Example to use trained model
|
||||
|
||||
In ['Example_Prediction'](ex_prediction.ipynb)
|
||||
|
||||
Reference in New Issue
Block a user