cat-vs-dog / src /config.py
eddydecena's picture
Add: Source
e4497d1
import os
DATASET_URL = 'https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_3367a.zip'
CACHE_DIR = os.getcwd()
CACHE_SUBDIR = 'data'
if not os.path.isdir(CACHE_SUBDIR):
os.mkdir(CACHE_SUBDIR)
DATASET_PATH = os.path.join(CACHE_DIR, CACHE_SUBDIR, 'PetImages')
IMAGE_SIZE = (180, 180)
BATCH_SIZE = 32
EPOCHS = 50