Run with a single Image

backseason / PoolNet

Code for our CVPR 2019 paper "A Simple Pooling-Based Design for Real-Time Salient Object Detection"

MIT License

628 stars 153 forks source link

Run with a single Image #32

Closed HHIR closed 4 years ago

HHIR commented 4 years ago

Good result, but the big problem of code is written in a very confusing way. for example, it is not easy to run code for just one image as the input and get one output!!! [ I mean not in the list file ] ... Please give some guidance about running code with a single image. ( not from the list)

thank

backseason commented 4 years ago

You can follow the functions in dataset/dataset.py to load the image file into torch.Tensor. And build the network as done in the solver.py file. Then just forward the loaded tensor into the network to get the predictions. It's easy and straightforward. Good luck!