Merical / Container_foreign_object_detection

A CNN based container foreign object detection algorithm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

A CNN based container foreign object detection algorithm


Languages

Language:Python 100.0%