Color-Classification-CNN
Color-Classification-CNN copied to clipboard
Number of epochs
Your implementation use 5 epochs. The paper says that they used 200000 iterations. The training set is composed of 7799 images. Using a batch size of 32 this would be around 820 epochs.