Diangle commited on
Commit
7fe2138
·
1 Parent(s): ce2ee37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,9 +60,9 @@ ft_visual_features_file_bin = DATA_PATH + '/dataset_v1_visual_features_database_
60
 
61
  #load database features:
62
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
63
- ft_visual_features_database = np.memmap(ft_visual_features_file, dtype='float32', mode='r')
 
64
 
65
- ft_visual_features_database = ft_visual_features_database[32:].reshape((ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
66
 
67
  database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
68
  database_df = pd.read_csv(database_csv_path)
 
60
 
61
  #load database features:
62
  ft_visual_features_database_bin = np.load(ft_visual_features_file_bin)
63
+ ft_visual_features_database = np.memmap(ft_visual_features_file, dtype='float32', mode='r', offset=128,
64
+ shape=(ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
65
 
 
66
 
67
  database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
68
  database_df = pd.read_csv(database_csv_path)