Skip to content

Instantly share code, notes, and snippets.

@yerkbn

yerkbn/cnn_import.py

Last active April 16, 2019 08:42
Show Gist options
  • Save yerkbn/7eca2783db07f1004f1277f8ab6f0dfe to your computer and use it in GitHub Desktop.
Save yerkbn/7eca2783db07f1004f1277f8ab6f0dfe to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cnn_import.py
import cv2
import numpy as np
import os
from random import shuffle
from tqdm import tqdm
TRAIN_DIR = '/home/yerkebulan/app/dev/projects/classification_research/dogs-vs-cats/train'
TEST_DIR = '/home/yerkebulan/app/dev/projects/classification_research/dogs-vs-cats/test1'
IMG_SIZE = 50
LR = 1e-3
EPOCHE = 10
MODEL_NAME = 'dog_vs_cat-{}-{}.model'.format(LR, '8conv-10epoche')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment