Skip to content

liuyi1989/TCGNet

Repository files navigation

TCGNet: Type-Correlation Guidance for Salient Object Detection

Requirements

  • PyTorch == 1.0.0
  • TorchVision == 0.2.1
  • CUDA 10.0 cudnn 7.2

Backbone Pretrained Model

Please download backbone pretrained model from https://1drv.ms/u/s!Aus8VCZ_C_33dKvqI6pBZlifgJk, and put it into backbone package.

Train

Please Run train.py

Infer

Please Run infer.py

Datasets

Please alter path in config.py and datasets.py

About

This is the code for TCGNet

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages