Sovit Ranjan Rath

Results 221 comments of 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. ![image_6](https://github.com/user-attachments/assets/35ead78b-70bd-4e0b-a2c3-b20b8c33daf5)

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....

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.