search

gouthamvgk / SuperGlue_training

This repository contains the training code for SuperGlue. Uses COCO dataset by generating random homographies.
Other
107 stars 24 forks source link

How to train superglue in my own datasets? #2

Closed zhanghua7099 closed 3 years ago

zhanghua7099 commented 3 years ago

This code seems to only support COCO 2017 dataset. How to train in another dataset? Can you give some advice?

gouthamvgk commented 3 years ago

Currently, the dataloader is coded in such a way that it accepts coco format only. You can easily convert your dataset into coco format. COCO format is as follow, DATASET_FOLDER: annotations/ instances_train2017.json -> Contains training images info instances_val2017.json -> Contains validation images info train2017/ contains the training images val2017/ contains the validation images

You can refer to the sample dataset at https://drive.google.com/drive/folders/1LrS2TKsgd1OVvrfObirDQBCaE4u2NSUN?usp=sharing if you want.(Sample dataset contains only train json and train images, have to create val json and images similarly)

zhanghua7099 commented 3 years ago

Thank you for your reply!

zhangsngood commented 1 year ago

Thank you for your reply!

This code seems to only support COCO 2017 dataset. How to train in another dataset? Can you give some advice?

Have you realized it？Now I want to use my own datasets to train the SuperGlue, but the error happens when training. So I want to ask if you have achieved it, and I want some suggestions from you.

Shafaq-S commented 10 months ago

Thank you for your reply!

This code seems to only support COCO 2017 dataset. How to train in another dataset? Can you give some advice?

Have you realized it？Now I want to use my own datasets to train the SuperGlue, but the error happens when training. So I want to ask if you have achieved it, and I want some suggestions from you.

Did any of you manage to train it on your own dataset? I have a similar problem where I'll be annotating my own data but unsure about the annotations format. Please refer to issue # 17.

Shuhul24 commented 10 months ago

@Shafaq-S Hey, did you get the solution on how to manage to generate such files for our respective custom dataset?

Shafaq-S commented 10 months ago

@Shafaq-S Hey, did you get the solution on how to manage to generate such files for our respective custom dataset?

I decided to not go with COCO format and defined my custom json annotations. Currently, I'm in the process of annotating my data. Still have to test the code with it.

Shuhul24 commented 9 months ago

@Shafaq-S Hey any updates regarding the json format of annotations? Need some help regarding the same.