Skip to content

leoromanovich/MechPart-Detection-solution

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MechPart-Detection-decision Pet Project

This is an automatic detection, segmentation, and counting of mechanical parts app based on Mask RCNN and Kivy.

If you want to reimplement this to your own task, please read tutorials for Mask_RCNN. For example Balloon example or Shapes example.

Common steps for making the decision:

  1. Prepare data for your task.
  2. Install necessary packages (Keras, Kivy, Numpy, Matplotlib, Opencv etc.)
  3. Use VIA annotator to annotate objects on images.
  4. Run Jupyter notebook, change paths to images, begin training of the neural network and validate results.
  5. Implement weights of your model (.h5 file) into the graphical interface. Just lay down .h5 file in root directory of this app.
  6. If you want to build your app for Windows or Linux, use .spec file and pyinstaller package.

Examples:

Raw Image

After NN processing

Interface example

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published