a5372935/Resnet_AOI_2classification

This is an exercise I used to do the AOI 2-classification with the ResNet CNN model (for multi-category training as well)

Resnet_AOI_2classification

This is an exercise I used to do the AOI 2-classification with the ResNet CNN model (for multi-category training as well)

This AOI is the board soldering judgment classification

Abstract

This project is based on Octave_Resnet_Tensorflow_keras, and completed by YOU-JEN SHIEH

Good smaple:

NG smaple:

AOI_epoch_acc and AOI_epoch_loss:

AOI_epoch_val_acc and AOI_epoch_val_loss:

Test data of resnet-18, I set Threshold 0.85

Test data of Oct_resnet-18, I set Threshold 0.85

Test data of resnet-18, I set Threshold 0.95

Test data of Oct_resnet-18, I set Threshold 0.95

Abstract

This project is based on OctConv-TFKeras, and completed by YOU-JEN SHIEH

Citation

@INPROCEEDINGS{9010309,
        author={Y. {Chen} and H. {Fan} and B. {Xu} and Z. {Yan} and Y. {Kalantidis} and M. {Rohrbach} and Y. {Shuicheng} and J. {Feng}},
        booktitle={2019 IEEE/CVF International Conference on Computer Vision (ICCV)}, 
        title={Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks With Octave Convolution}, 
        year={2019},
        volume={},
        number={},
        pages={3434-3443},}
}