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

Pytorch implementation of Amulet(ICCV 2017).
https://github.com/xsxszab/amulet-pytorch

pytorch salient-object-detection

Last synced: 3 months ago
JSON representation

Pytorch implementation of Amulet(ICCV 2017).

• Host: GitHub
• URL: https://github.com/xsxszab/amulet-pytorch
• Owner: xsxszab
• Created: 2020-01-26T14:50:36.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2020-02-09T03:02:12.000Z (over 5 years ago)
• Last Synced: 2025-04-05T19:44:25.764Z (6 months ago)
• Topics: pytorch, salient-object-detection
• Language: Python
• Homepage:
• Size: 23.4 KB
• Stars: 3
• Watchers: 1
• Forks: 1
• Open Issues: 2
• Metadata Files:
  • Readme: readme.md

Awesome Lists containing this project

README

          
## Amulet net for salient object detection

A Pytorch implementation for Amulet(ICCV 2017).

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

(Currently bugs already known are fixed. If you found any bugs

 in this paper reproduction, you can open an issue, and I will

 fix it when I have time, thanks!)