Scene-Graph-Benchmark.pytorch icon indicating copy to clipboard operation
Scene-Graph-Benchmark.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon KaihuaTang

Finetune on custom dataset?

Open DesaleF opened this issue 2 years ago • 4 comments

Hi, Dear Author! First thank you for sharing your work. I explore the code and also tested the model on random images. I found your model very useful for my project. However, I have object labels and relationship(predicates) which doesn't exist in visual-gnome dataset. So I decided to finetune your model using my custom dataset. Can you help me with the steps I should follow to use your code for my custom dataset?

Thank you!

DesaleF avatar May 16 '22 08:05 DesaleF

Hi @DesaleF! Did you figure out how to do it? I can retrain the custom dataset from scratch but did you figure out how to fine tune it on the top of pretrained model (on VG dataset)?

bibekyess avatar Aug 12 '22 09:08 bibekyess

Maybe this helps: #180

bibekyess avatar Dec 23 '22 07:12 bibekyess

Maybe this helps: #180

Hello, I have been learning about scene maps recently, but I am not sure how to create my own training set. If you know how to create it and let me know, I would be very grateful

majianbo3 avatar May 13 '24 08:05 majianbo3

Maybe this helps: #180

Hello, I have been learning about scene maps recently, but I am not sure how to create my own training set. If you know how to create it and let me know, I would be very grateful

Hello, do you now know how to create dataset?

zzzcccxx avatar Jun 07 '24 01:06 zzzcccxx