{"id":28149933,"url":"https://github.com/zal0302/cii","last_synced_at":"2025-07-16T05:36:27.378Z","repository":{"id":54478301,"uuid":"503275374","full_name":"zal0302/CII","owner":"zal0302","description":"The official PyTorch implementation of IEEE Transactions on Image Processing 2021 paper \"Rethinking the U-shape Structure for Salient Object Detection\"","archived":false,"fork":false,"pushed_at":"2022-12-01T02:52:21.000Z","size":75,"stargazers_count":20,"open_issues_count":1,"forks_count":2,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-05-15T02:14:56.916Z","etag":null,"topics":["object-segmentation","pytorch","saliency","saliency-detection","saliency-maps","saliency-prediction","salient-object-detection","u-net"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/zal0302.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2022-06-14T08:36:25.000Z","updated_at":"2025-04-30T02:47:44.000Z","dependencies_parsed_at":"2023-01-23T12:00:25.184Z","dependency_job_id":null,"html_url":"https://github.com/zal0302/CII","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/zal0302/CII","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/zal0302%2FCII","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/zal0302%2FCII/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/zal0302%2FCII/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/zal0302%2FCII/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/zal0302","download_url":"https://codeload.github.com/zal0302/CII/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/zal0302%2FCII/sbom","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":265485016,"owners_count":23774417,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["object-segmentation","pytorch","saliency","saliency-detection","saliency-maps","saliency-prediction","salient-object-detection","u-net"],"created_at":"2025-05-15T02:14:55.838Z","updated_at":"2025-07-16T05:36:27.336Z","avatar_url":"https://github.com/zal0302.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"## Rethinking the U-shape Structure for Salient Object Detection\n\n### This is the official PyTorch implementation of our TIP 2021 [paper](https://mftp.mmcheng.net/Papers/21TIP-CII.pdf).\n\n## Prerequisites\n\n- [Pytorch 0.4.1+](http://pytorch.org/)\n- [torchvision](http://pytorch.org/)\n\n\n## Usage\n\n### 1. Clone the repository\n\n```shell\ngit clone https://github.com/zal0302/CII.git\ncd CII/\n```\n\n### 2. Download the datasets\n\nDownload the following [datasets for testing](https://drive.google.com/file/d/1jIL3Yvly4l4l_OggljjreD87pJIm_6rm/view?usp=sharing) and unzip them into `data` folder.\n\n### 3. Download the pre-trained models for CII and backbone\n\nDownload the following pre-trained models for CII with [ResNet50 backbone](https://drive.google.com/file/d/1JcePr4FwWMedhFHeClYF1v_MIYwJGOF0/view?usp=sharing) and [ResNet18 backbone](https://drive.google.com/file/d/1DL860taDrmDUv-Am49AQZsdcF4Ey2-2t/view?usp=sharing) into `saved/models` folder. \n\n### 4. Test\n\nFor all datasets testing used in our paper for ResNet50 backbone:\n\n```shell\npython test.py -r saved/models/cii.pth -c saved/models/config.json\n```\n\nand for ResNet18 backbone:\n\n```shell\npython test.py -r saved/models/cii_res18.pth -c saved/models/config_resnet18.json\n```\n\nAll results saliency maps will be stored under `saved/results` folders in .png formats.\n\n### 5. Pre-computed results and evaluation results\n\nYou may refer to this repo for results evaluation: [SalMetric](https://github.com/Andrew-Qibin/SalMetric).\n\nWe provide the pre-computed saliency maps and evaluation results for [ResNet50 backbone](https://drive.google.com/file/d/11Uj2-qNDyASrfvdXj2uE9Zm7xiYYwNEM/view?usp=sharing) and [ResNet18 backbone](https://drive.google.com/file/d/1Q53oKWTNA9KznWmbXGm2IhY_2yeYvF1E/view?usp=sharing).\n\n### 6. Contact\n\nIf you have any questions, feel free to contact me via: `liuzhiang(at)mail.nankai.edu.cn`.\n\n\n### If you think this work is helpful, please cite\n\n```latex\n@article{liu2021rethinking,\n  title={Rethinking the U-Shape Structure for Salient Object Detection},\n  author={Liu, Jiang-Jiang and Liu, Zhi-Ang and Peng, Pai and Cheng, Ming-Ming},\n  journal={IEEE Transactions on Image Processing},\n  volume={30},\n  pages={9030--9042},\n  year={2021},\n  publisher={IEEE}\n}\n```\n```latex\n@article{liu2022poolnet+,\n  title={Poolnet+: Exploring the potential of pooling for salient object detection},\n  author={Liu, Jiang-Jiang and Hou, Qibin and Liu, Zhi-Ang and Cheng, Ming-Ming},\n  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},\n  year={2022},\n  publisher={IEEE}\n}\n```\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fzal0302%2Fcii","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fzal0302%2Fcii","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fzal0302%2Fcii/lists"}