From 50b98fcaa1edc658826da500c87955349e55888a Mon Sep 17 00:00:00 2001 From: CJC Date: Thu, 6 Apr 2017 23:13:33 -0700 Subject: [PATCH] separate ui to another branch --- .gitignore | 1 + models/finetuned_resnet.py | 2 +- predict.py | 6 +++--- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 5721651..98ef253 100644 --- a/.gitignore +++ b/.gitignore @@ -4,3 +4,4 @@ data/ *.h5 /logs .DS_Store +__pycache__ diff --git a/models/finetuned_resnet.py b/models/finetuned_resnet.py index 74a38b2..9852283 100644 --- a/models/finetuned_resnet.py +++ b/models/finetuned_resnet.py @@ -1,7 +1,7 @@ import os from keras.models import Model from keras.layers import Dense, Flatten, Dropout -from .resnet50 import ResNet50 +from resnet50 import ResNet50 N_CLASSES = 101 IMSIZE = (216, 216, 3) diff --git a/predict.py b/predict.py index 99d0646..73fe91b 100644 --- a/predict.py +++ b/predict.py @@ -1,8 +1,8 @@ import os import numpy as np -from .utils.UCF_utils import two_stream3_generator, two_stream18_generator -from .models.two_stream import two_stream_model -from .utils.OF_utils import stack_optical_flow +from utils.UCF_utils import two_stream3_generator, two_stream18_generator +from models.two_stream import two_stream_model +from utils.OF_utils import stack_optical_flow import cv2 import random from scipy.misc import imresize