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:
|
## To do list:
|
||||||
- [x] Import model detection (SSD & YOLO3)
|
- [x] Import model detection (SSD & YOLO3)
|
||||||
- [x] Model Panel Detection
|
- [x] Model Panel Detection
|
||||||
- [ ] Model Soiling Fault Detection
|
- [x] Model Soiling Fault Detection
|
||||||
- [ ] Model Diode Fault Detection
|
- [x] Model Diode Fault Detection
|
||||||
- [ ] Model Other Fault Detection
|
- [ ] Model Other Fault Detection
|
||||||
|
|
||||||
|
|
||||||
@@ -20,7 +20,10 @@
|
|||||||
* OpenCV
|
* OpenCV
|
||||||
* Beautiful Soup 4.x
|
* 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
|
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
|
### 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
|
## 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