Skip to content

A question above negative loss calculation #2

Description

@sunshineatnoon

Hi, Thanks for this awesome work, I'm trying to train it on my own dataset, so I am exploring how to organize my data. I notice that for unmatched images, you calculate their matching loss by rolling a batch of images(https://github.com/ignacio-rocco/ncnet/blob/master/train.py#L137):

    batch['source_image']=batch['source_image'][np.roll(np.arange(b),-1),:] # roll

I don't quite get why moving each sample one position forward creates unmatched pairs, how do you organize your data? Also, is the set information in the csv files useful during training? Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions