Sovit Ranjan Rath
Sovit Ranjan Rath
Hi. Can you please check that you have PyTorch version at least 1.12.1 and ONNX version 1.12.0? I trained Faster RCNN ResNet FPN V2 with PyTorch 2.2.0 and was able...
Hello @merbenf I have not integrated SAHI into the project yet. I will add that to my TODO list.
I will be working on this next.
@joshua-atolagbe I have SAHI image inference for all the pretrained torchvision models with almost all the arguments similar to `inference.py`. The code is in the `sahi_inference.py` file. You can find...
@joshua-atolagbe Thanks. Just for some clarification. It is the inference script. It will give you the FPS on the folder of images (the directory path that you will pass that...
@joshua-atolagbe Just for addition. These are the kind of results that you can expect with the current pipeline with the COCO pretrained Faster RCNN ResNet50 FPN V2. 
Just an update. Updated the `sahi_inference.py` script for custom trained models as well. Now you can use not just Torchvision pretrained models, but also custom trained weights with the script....
Thanks.
Hello. Please follow this [inference](https://github.com/sovit-123/fasterrcnn-pytorch-training-pipeline?tab=readme-ov-file#Inference) section of README. It has instructions to carry out predictions in all formats. Simply, this is an example command for image inference: python inference.py --input...
Will take a look into it ASAP. Need to check the code. Maybe a threshold issue.