Fix issue with googlenet module by using relative imports on Python 3.

darrengarvey · darrengarvey · commit 36481fdfab0c · 2015-06-30T22:32:30.000+01:00
This commit fixes an issue with importing `sklearn_theano.feature_extraction` due to relative paths not being used within that module. The commit fixes the example on the sklean-theano homepage of recognising cats and dogs in photo [1], clearly a primary use-case of any such machine learning library. Errors fixed (listed for searchability): - No module named 'caffemodel' - No module named 'googlenet_class_labels' - No module named 'googlenet_layer_names' [1] http://sklearn-theano.github.io/auto_examples/plot_multiple_localization.html
diff --git a/sklearn_theano/feature_extraction/caffe/googlenet.py b/sklearn_theano/feature_extraction/caffe/googlenet.py
@@ -5,11 +5,11 @@
 
 from sklearn.externals import joblib
 from ...datasets import get_dataset_dir, download
-from caffemodel import _parse_caffe_model, parse_caffe_model
+from .caffemodel import _parse_caffe_model, parse_caffe_model
 import os
 from ...utils import check_tensor, get_minibatch_indices
-from googlenet_class_labels import get_googlenet_class_label
-from googlenet_layer_names import get_googlenet_layer_names
+from .googlenet_class_labels import get_googlenet_class_label
+from .googlenet_layer_names import get_googlenet_layer_names
 from sklearn.base import BaseEstimator, TransformerMixin
 import theano
 import numpy as np
