/defect-segment-detection-ui

UI of Detection and Segmentation in Powder Spreading Process of Magnetic Material Additive Manufacturing.

Primary LanguagePythonMIT LicenseMIT

How about this repo?

  • Here is the repository for the UI of the defect detection and segmentation with digital image processing with N.C.K.U. CSIE course.
  • The project scope is Detection and Segmentation in the Powder Spreading Process of Magnetic Material Additive Manufacturing.
  • If you want to know the algorithm, please refer to the following link.

Result

image image

How to run?

  1. install requirements
pip install -r requirements.txt
  1. cd to the project directory
cd defect-segment-detection-ui/ui
  1. run UI
python  defect-detection2.py
  1. download the model with release and move it to the model directory
# move to yolov7
mv yolov7_best.pt ../script/yolov7

# move to unet
mv unet_best.pth ../script/unet