search

ttharden / Keyframe-Extraction-for-video-summarization

19 stars 3 forks source link

采用torch计算K均值聚类。 #2

Open glide-the opened 1 year ago

glide-the commented 1 year ago

you are using a cpu of NumPy operations, which can be inefficient. Here's a modified version of your code that uses PyTorch tensors throughout the process:

ba9869aeb9b5d02cac83aa1af817c0e