Autor: Mihai Nan
În această problemă trebuie să implementați un model de clasificare a imaginilor capabil să distingă între imagini de artă reală și imagini generate de inteligență artificială.
Setul de date este organizat într-un director cu imagini și fișiere CSV, iar performanța modelului va fi evaluată folosind F1-score.
Setul de date conține două tipuri de imagini:
0)1)Aceste imagini sunt disponibile în directorul images.
Fișierele asociate:
train.csv – pentru antrenaretest.csv – pentru predicțiiColoane:
images/img_000123.jpg)0 pentru artă reală, 1 pentru artă generată de AIColoane:
Fără etichete — acestea trebuie prezise de model.
Modelul trebuie să învețe să clasifice fiecare imagine în una dintre următoarele clase:
Evaluarea se face folosind F1-score pe predicțiile trimise pentru setul de test.
Formula:
Fișierul CSV trebuie să conțină exact două coloane:
| Coloană | Tip | Descriere |
|---|---|---|
SampleID | integer | ID-ul imaginii din test |
Label | integer | Predicția (0 sau 1) |
| SampleID | Label |
|---|---|
| 101 | 0 |
| 102 | 1 |
| 103 | 1 |
Pentru această problemă se folosește un set de date public disponibil pe Kaggle.