CenterNet (Objects as Points) demo

tugstugi / dl-colab-notebooks

Try out deep learning models online on Google Colab

1.72k stars 464 forks source link

CenterNet (Objects as Points) demo #50

Closed melaniihariono closed 2 years ago

melaniihariono commented 2 years ago

Hello. Iam trying to run the centernet demo. When I change the model to Centernet hourglass, then I found this. Can anyone tell me what is this? Thanks!

tugstugi commented 2 years ago

dowload first the correct weight:

model_name = 'ctdet_coco_hg.pth'
if not exists(model_name):
!pip install -q gdown
!gdown 'https://drive.google.com/uc?id=1cNyDmyorOduMRsgXoUnuyUiF6tZNFxaG'

change the model architecture to hourglass:

opt.load_model = model_name
opt.arch = 'hourglass'

It should work now as intended.