search

shreyaskamathkm / FTNet

Pytorch implementation of FTNet for Semantic Segmentation on SODA, SCUT Seg, and MFN Datasets
Other
17 stars 3 forks source link

Inference on single image\folder #2

Open iariav opened 2 years ago

iariav commented 2 years ago

Hi, thanks for the great work. I want to test some of the pre-trained models on my own data. is there some simple way to run a pre-trained model on a folder of images? or on a single image? thanks

Harsh188 commented 6 months ago

This would be quite useful.

shreyaskamathkm commented 5 months ago

Currently we are in the process of updating the repo with the latest packages. We should have this functionality soon.

Harsh188 commented 5 months ago

Is there a quick solution to running inference using fine-tuned ckpt? I want to visualize results on thermal data that I've captured.

shreyaskamathkm commented 5 months ago

I have something on the 'develop' branch. But it's work in progress. Should be able to publish it once I refactor it.

shreyaskamathkm commented 4 months ago

The develop branch has been updated to support inferring on a folder of images. Please use that branch and follow the instructions. Please ensure you have the downloaded rev-2 models from the box. The old model checkpoints will not work.