Skip to content
This repository was archived by the owner on Jun 11, 2026. It is now read-only.

What does the "centroids" mean? #12

Description

@9p15p

Hi, Thank you for your excellent work!

I want to know what the 'annotation_centroids.npy' and the centroids means.
To be specific,I can't find something in paper related to torch.argmin(torch.sqrt(torch.sum((img.unsqueeze(1) - centroids) ** 2, 2)), 1) in function 'rgb2class'.

Best wishes.

Here is my question in Chinese,which is consistent with the above:

'centroids'这个变量以及在dataset中的 'annotation_centroids.npy' 发挥了什么作用呢,我在论文中没有看到对应的信息。
关于'rgb2class'函数中的 torch.argmin(torch.sqrt(torch.sum((img.unsqueeze(1) - centroids) ** 2, 2)), 1) 具体含义麻烦您能解释一下吗

祝好。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions