Skip to content

clovermini/MVIT_metal_datasets

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

24 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

MVIT_metal_datasets

Provided by Machine Vision and Industrial Testing Laborator (MVIT Lab).

πŸ“Œ Overview

We release two metal surface defect datasets with instance-level pixel annotations: Casting Billet and Steel Pipe, as well as a Medium and Heavy Plate Surface Defect Dataset annotated in YOLO format.

πŸ—ƒοΈ Datasets

1. Casting Billet Dataset

  • Images: 1,060 (780 defective)
  • Resolution: 96Γ—106 to 3,228Γ—492
  • Defect Types:
    • Scratch
    • Weld slag
    • Cutting opening
    • Water slag mark
    • Slag skin
    • Longitudinal crack

2. Steel Pipe Dataset

  • Images: 1,227 (554 defective)
  • Resolution: 728Γ—544 (fixed)
  • Defect Types:
    • Warp
    • External fold
    • Wrinkle
    • Scratch

3. Medium and Heavy Plate Surface Defect Dataset

  • Images: 680 (480 defective)
  • Resolution: 256Γ—256 (fixed)
  • Defect Types:
    • Inclusions (In)(120 samples)
    • Blocky Scale (Bs)(120 samples)
    • Striated Scale (Ss)(120 samples)
    • Foreign Object Embedding (Foe)(120 samples)

✏️ Annotation Process

  1. AI Pre-segmentation
    Leverage SAM's predictive interface to perform batch automatic segmentation, generating initial masks based on the provided bounding box annotations and images.

  2. Expert Refinement
    1). Identification of Suboptimal Segmentation:
    Review the initial masks to identify suboptimal segmentation results through human assessment.

    2). Interactive Refinement:
    For suboptimal results, use SAM's interactive segmentation by iteratively adding:

    • Positive sample points to guide the identification of the target region.
    • Negative sample points to exclude interference regions.
      Continuously update the segmentation results in real-time until the desired accuracy is achieved.

    3). Post-processing:

    • Perform threshold-based segmentation using optimal thresholds for the specific dataset.
    • Apply morphological operations, including opening and closing, to smooth edges, eliminate noise, fill holes, and perform other enhancements.

Label Process

πŸ–ΌοΈ Samples

Dataset Samples Dataset Samples

πŸ“₯ Download

Download Link(baiduyun) | Alternative links(google drive)

πŸ“œ Citation

πŸ“§ Contact

For dataset inquiries or collaboration opportunities: πŸ“§ [email protected] πŸ“§ [email protected]


Maintained by MVIT Lab @ Collaborative Innovation Center of Steel Technology, University of Science and Technology Beijing

About

Machine Vision and Industrial Application Laboratory (MVIA Lab)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •