Collaborative-Optimization

This is the official repository for our paper, "CONet: A Collaborative Dual-Network Optimization Framework for Rail Surface Defect Detection".

Architecture and Details:

Architecture of the CONet:

Architecture of the FDPNet：

Architecture of the ASSNet：

Results

Comparison of Rail Defect Detection Results

CONet ablation experiments

FDPNet ablation experiments

ASSNet ablation experiments

Weights and Visualization Results

You can download the pre-trained weights and visualization results from the following link:

Link: Baidu Netdisk: 链接: https://pan.baidu.com/s/19XHRLgCTRcK9S5g-lwNhwA

Base Framework and Dataset

The base framework and the dataset used in our work can be obtained from the following repository: 链接: https://pan.baidu.com/s/19XHRLgCTRcK9S5g-lwNhwA

Code Release

The complete source code will be released upon the acceptance of our paper. Stay tuned!

Contact

Please drop me an email for any problems or discussion:1342982253@qq.com.