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¶
Homepage: Casting Product Quality
Licenses: Attribution-NonCommercial-NoDerivatives 4.0 International
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 |