Kaggle Casting Quality

Attribute

Value

pretty_name

Kaggle Casting Quality

annotations_creators

language_creators

languages

[]

licenses

CC BY-NC-ND 4.0

multilinguality

size_categories

1K<n<10K

source_datasets

task_categories

image-classification

task_ids

paperswithcode_id

Dataset Description

Dataset Summary

Dataset for the automatic identification of casting defects.

Download and prepare data

Download the data directly from Kaggle and extract it. Replace {PATH_TO_DATA} below with the location of the data. Use the following code to load it:

from squirrel_datasets_core.datasets.kaggle_casting_quality.driver \
    import RawKaggleCastingQualityDriver
iter_train = RawKaggleCastingQualityDriver("{PATH_TO_DATA}/casting_data/casting_data")\
    .get_iter("train")
iter_test = RawKaggleCastingQualityDriver("{PATH_TO_DATA}/casting_data/casting_data")\
    .get_iter("test")

Dataset Structure

Data Instances

A sample from the training set is provided below:

{
    'url': '{PATH_TO_DATA}/casting_data/casting_data/test/ok_front/cast_ok_0_9996.jpeg',
    'label': 1,
    'image': array(...)
}

Dataset Schema

  • img: A numpy array containing the 300x300 RGB image.

  • label: 1 for ok front and 0 for defect front.

Data Splits

name

train

test

kaggle-casting

6633

715