מחבר: Mihai Nan
In this problem, you must implement an image classification model capable of distinguishing between real artwork and AI-generated artwork.
The dataset is organized into a directory containing images and CSV files, and the model’s performance will be evaluated using the F1-score.
The dataset contains two types of images:
0)1)These images are stored in the images directory.
Associated files:
train.csv – used for trainingtest.csv – used for predictionsColumns:
images/img_000123.jpg)0 for real art, 1 for AI-generated artColumns:
No labels — these must be predicted by the model.
The model must learn to classify each image into one of the following classes:
Evaluation is performed using the F1-score computed on the predictions submitted for the test set.
Formula:
The submission CSV file must contain exactly two columns:
| Column | Type | Description |
|---|---|---|
SampleID | integer | Test image ID |
Label | integer | Prediction (0 or 1) |
| SampleID | Label |
|---|---|
| 101 | 0 |
| 102 | 1 |
| 103 | 1 |
This problem uses a public dataset available on Kaggle.