paucarre / tiefvision

End-to-end deep learning image-similarity search engine
Apache License 2.0
765 stars 186 forks source link

bounding box models? #61

Closed tpeng closed 7 years ago

tpeng commented 7 years ago

hi, in bboxlib.lua it loads several models by:

local function loadLocator(index)
  return torch.load(tiefvision_commons.modelPath('locatorconv-' .. index .. '.model'))
end

local function loadClassifier()
  return torch.load(tiefvision_commons.modelPath('classifier.model'))
end

local function loadEncoder()
  return torch.load(tiefvision_commons.modelPath('encoder.model'))
end

do you have pre-trained models so i can download?

thanks,

plippe commented 7 years ago

Hi Terry,

We wouldn't have any models to share, but they are quite easy to generate.

I would advise you to look at bin/unsupervised_similarity. This file has everything you need to get started.

Good luck

tpeng commented 7 years ago

thanks Plippe!