search

zzzxxxttt / pytorch_simple_CenterNet_45

A simple pytorch implementation of CenterNet (Objects as Points)
308 stars 63 forks source link

Inference #16

Closed PengZH99 closed 4 years ago

PengZH99 commented 4 years ago

if i want to inference a single image and show the results on the image,how can I achieve it quickly?I would appreciate it if you can share the inference code.thank u

zzzxxxttt commented 4 years ago

Just replace the dataloader with your own image loader in test.py, and add a few lines of code to plot the results.

PengZH99 commented 4 years ago

Just replace the dataloader with your own image loader in test.py, and add a few lines of code to plot the results.

I've been trying all the day,but i can't achieve it, it's so hard for me.can u help me ,thank u!

zzzxxxttt commented 4 years ago

I've added a demo.py in the latest commit.

PengZH99 commented 4 years ago

I've added a demo.py in the latest commit.

Thank u so much!