https://github.com/xsxszab/cpd-pytorch

Pytorch implementation of CPD net(CVPR2019) for salient object detection.
https://github.com/xsxszab/cpd-pytorch

Last synced: 6 months ago
JSON representation

Pytorch implementation of CPD net(CVPR2019) for salient object detection.

• Host: GitHub
• URL: https://github.com/xsxszab/cpd-pytorch
• Owner: xsxszab
• Created: 2020-01-28T02:42:56.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2020-02-09T10:07:25.000Z (over 5 years ago)
• Last Synced: 2025-02-11T18:45:56.025Z (8 months ago)
• Language: Python
• Size: 22.5 KB
• Stars: 4
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: readme.md

Awesome Lists containing this project

README

          
## CPD net for salient object detection

Pytorch implementation of CPD(CVPR 2019), some codes(part of

network structure definition) are from repository

 https://github.com/wuzhe71/CPD ,this project is for practice.

 

 ### Usage

* Train

    1. Put corresponding dataset in ./input/

        * training images(RGB, jpg format): ./input/train/raw/

        * training masks(gray, png format): ./input/train/mask/

        * validation images(RGB, jpg format): ./input/test/raw/

        * validation masks(gray, png format): ./input/test/mask/

    2. Run train.py, if you want to change some parameters,

    see train.py for detail.

* Inference

    1. Put inference data in ./inference/

        * inference images(RGB, jpg format): ./inference

    2. Run inference.py, output saliency maps will be in

    ./output directory.

### Requirements

Original running environment:

* Python 3.7.5

* Pytorch 1.3.1

* TorchVision 0.2.1

* pillow 7.0.0

Detailed requirements are listed in requirements.txt.