search

Media-Smart / vedadet

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

How to see an infer DEMO? #74

Open ModynAI opened 2 years ago

ModynAI commented 2 years ago

Hi, First of all i wanna tank you for your great work.

i followed your instruction on Inference. first, i installed pytorch and cuda and then git clone the repo and then set root dir. after that i installed all the dependencies and libraries! after that I've download the pretrain weights to my workstation. then i went into the this: https://github.com/Media-Smart/vedadet/blob/main/configs/infer/tinaface/tinaface_r50_fpn_bn.py

and changed the weight pass to the downloaded .pth file from your google drive ( https://drive.google.com/file/d/1zZ88-H3721F4V9-9NVpHFrbcpACflrHU/view?usp=sharing).

at the end i just used an image to test for a demo using this command CUDA_VISIBLE_DEVICES="0" python tools/infer.py configs/infer/tinaface/tinaface_r50_fpn_bn.py /content/13.jpg

Am i doing sth wrong? is it necessary to train the model?

Thanks Best regards

ModynAI commented 2 years ago

@mike112223 @mileistone @hxcai @DarthThomas

gdiasbruno commented 1 year ago

There is any answer to this?