Media-Smart / vedadet

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

pretrained models retinanet_r50.pth #70

Open fisakhan opened 3 years ago

fisakhan commented 3 years ago

While running inference immediately after installation of vedadet, I received the following error.

File "tools/infer.py", line 93, in main() File "tools/infer.py", line 75, in main engine, data_pipeline, device = prepare(cfg) File "tools/infer.py", line 31, in prepare load_weights(engine.model, cfg.weights.filepath) File "/home/tf/Projects/torch/vedadet-main/vedacore/misc/checkpoint.py", line 261, in load_weights state_dict = _load_checkpoint(filepath, map_location) File "/home/tf/Projects/torch/vedadet-main/vedacore/misc/checkpoint.py", line 135, in _load_checkpoint raise IOError(f'{filepath} is not a file') OSError: retinanet_r50.pth is not a file

Can you please share the retinanet_r50.pth ? Can you please make the inference part easy to use and self-contained? It will make the life others super easy and this repository super useful, super popular.