Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

Pretrained #14

Closed CengizhanYurdakul closed 3 years ago

CengizhanYurdakul commented 3 years ago

I could not find the pretrained model. Where can I download it?

hxcai commented 3 years ago

@CengizhanYurdakul Hi, which pretrained model do u want?

hxcai commented 3 years ago

@CengizhanYurdakul You can download models from mmdetection and you need to extract weights to save a new file like following

model = torch.load(mmdetection_model_file)
weights = model['state_dict']
torch.save(weights, vedadet_weight_file)
CengizhanYurdakul commented 3 years ago

@CengizhanYurdakul Hi, which pretrained model do u want?

I found this repo from PaperswithCode. I am looking for state of the art face detection model.

hxcai commented 3 years ago

@CengizhanYurdakul Please see this for details.