Diangle commited on
Commit
ce2ee37
·
1 Parent(s): 368f9d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -60,8 +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
- shape=(ft_visual_features_database_bin.shape[0], ft_visual_features_database_bin.shape[1]*8))
 
65
 
66
  database_csv_path = os.path.join(DATA_PATH, 'dataset_v1.csv')
67
  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')
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)