Train Faster R-CNN, Cascase R-CNN, and RetinaNet models on Tensorpack and MMDetection with thermal images for failure detection.
-
Prepare dataset
- Extract thermal(There’re 115 films in total, 250 images contained in each of them.)
- Data filtering: bluring, and adjacent similiar images.
- Labeling: LabelImg, save annotaions with Pascal VOC data format. Used labels refer to 'data_engineering'.
- Test set: pick out 10(including all labels) films for final test.
- Augmentation: find the needed scripts in the folder 'augmentation'.
-
Setup training frameworks
-
Install MMDetection
Relatted Adaptation: data foler; class names; used detectior; pre-trained model; results folder.
-
Install Tensorpack
Relatted Adaptation: data foler; class names; used detectior; pre-trained model; results folder.
-
-
Start training
- Scripts provided. For ease of changing detectors, use some shell scripts.
- Monitoring the process with tensorboard(Tensorpack) and built-in tools(MMDetection).
-
Evaluation/Test (scripts provided).
-
Deployment:
On platform Nvidia Jetson AGX Xavier(Nvidia Xavier).