/Container_foreign_object_detection

A CNN based container foreign object detection algorithm

Primary LanguagePython

Container_foreign_object_detection

A CNN based container foreign object detection algorithm

dataset.py: Generating training and test datasets with patches

get_patches.py: Generating patches with background substraction

detect_lock.py: Demo, please change the pathes to match your own dirs

region_clasifier.py: Training the detection network

tools.py: Declaration of the model

IoU.py: Warped an IoU