DPS100 / DataSynthesizer

Synthesize images to be used for a supervised image classifier
0 stars 0 forks source link

Design model architecture #4

Open DPS100 opened 5 days ago

DPS100 commented 5 days ago

Figure out type and parameters of model to use for each dataset.

Bashir1999 commented 14 hours ago

Deep Preprocessing was not needed. PyTorch can be used to implement CNN cz of ease of use. I have added the notebook for training models. Take a look at the code and comment if you find issues. The dataset is organized into files: train -> is a folder, test -> folder, syn_img -> folder, inside each folder, there are three other folders each containing one class, apple folder, banana folder, and orange folder