Skip to content

Edit data processing files

Jump to bottom
Amy Kendig edited this page Jul 25, 2019 · 1 revision

We made some changes to the data processing files (used in this tutorial), so that only color images split into 80% train/20% test would be used. For more configurations, see Mohanty et al. 2016 and their github.

# Edit create_data_distribution.py: 
# line 63: INPUT_FOLDER +"/color"
# line 86: train_probs = [80]
Clone this wiki locally