search

WenbinLee / CovaMNet

The Pytorch code of "Distribution Consistency based Covariance Metric Networks for Few-shot Learning", AAAI 2019.
Other
56 stars 10 forks source link

Custom dataset #2

Closed MaleshwarSastri closed 4 years ago

MaleshwarSastri commented 4 years ago

HI, I am working on my custom logo detection dataset.Can you guide me on how to train and test(detection) my model.Mainly im not able to understand the detection part.Kindly provide more details,also mention on how to use 5 shot algorithm

WenbinLee commented 4 years ago

Hi, thanks for interesting in our work. Because our code is specifically designed for image classification, it's a little different from the detection task. I think maybe you can replace the classifier module of our model with a detection module. Hope this can help you.