VideoBERT icon indicating copy to clipboard operation
VideoBERT copied to clipboard

Published 20 hours ago • verified publisher icon ammesatyajit

AttributeError: 'MiniBatchKMeans' object has no attribute 'cluster_centers_'

Open harshraj32 opened this issue 3 years ago • 4 comments

Traceback (most recent call last): File "VideoBERT/VideoBERT/I3D/minibatch_hkmeans.py", line 30, in main() File "VideoBERT/VideoBERT/I3D/minibatch_hkmeans.py", line 26, in main hkmeans(root, prefix, 4, 12, batch_size, 15, save_dir, 'vecs', centroid_dir) File "/usr/local/lib/python3.7/dist-packages/hkmeans_minibatch/hkmeans.py", line 99, in hkmeans hkmeans_recursive(root, prefix, h, k, batch_size, epochs, save_dir, save_prefix, centroid_dir) File "/usr/local/lib/python3.7/dist-packages/hkmeans_minibatch/hkmeans.py", line 85, in hkmeans_recursive centroids, labelled_data = minibatch_kmeans(root, prefix, k, batch_size, epochs) File "/usr/local/lib/python3.7/dist-packages/hkmeans_minibatch/hkmeans.py", line 42, in minibatch_kmeans return kmeans.cluster_centers_, labelled_data AttributeError: 'MiniBatchKMeans' object has no attribute 'cluster_centers_'

the line i ran without any changes : !python3 VideoBERT/VideoBERT/I3D/minibatch_hkmeans.py -r /content/VideoBERT/VideoBERT/data/features -b 1 -s /content/VideoBERT/VideoBERT/data/kmean_vectors -c /content/VideoBERT/VideoBERT/data/centroids -p hkm

harshraj32 avatar May 10 '21 10:05 harshraj32

Your batch size seems to be 1. Ideally, it should be a multiple of how many vectors are in each file containing the feature vectors. This probably is the problem, since I had files with 20 vectors each, and I used a batch size of 500.

ammesatyajit avatar May 10 '21 17:05 ammesatyajit

If you have a batch size of 1, the kmeans will not be able to complete because by definition it requires multiple examples to update the centroids.

ammesatyajit avatar May 10 '21 18:05 ammesatyajit

also should this be the output in label.json file we are creating, looks like hkmeans is predicting all zero values for the files

"QU1Y4ogmSDE": [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ],

harshraj32 avatar May 12 '21 04:05 harshraj32

Has the problem been solved? I also had the same problem

chenjianqiang199526 avatar Jun 20 '22 11:06 chenjianqiang199526