AI-Generated-Image-Classification

Using PyTorch Vision to build an AI model that classifies and labels AI-generated images with high accuracy

Data:

• Image Data from Kaggle 11 GB

Issues:

• Traceback (most recent call last): File "", line 1, in File "/Library/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "/Library/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/spawn.py", line 126, in _main self = reduction.pickle.load(from_parent) AttributeError: Can't get attribute 'CustomImageDataset' on <module 'main' (built-in)>
• Issue with DataLoader Some images in between get shape of (1, 64, 64) instead of (3, 64, 64)