https://github.com/jahongir7174/ctdnet
https://github.com/jahongir7174/ctdnet
Last synced: 8 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/jahongir7174/ctdnet
- Owner: jahongir7174
- Created: 2024-04-08T22:25:56.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2024-04-12T07:14:29.000Z (over 1 year ago)
- Last Synced: 2024-04-12T15:06:50.173Z (over 1 year ago)
- Language: Python
- Size: 61 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
Complementary Trilateral Decoder for Fast and Accurate Salient Object Detection
### Installation
```
conda create -n PyTorch python=3.11
conda activate PyTorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
pip install opencv-python==4.5.5.64
pip install tqdm
pip install timm
```
### Train
* Configure your dataset path in `main.py` for training
* Run `python main.py --train` for Single-GPU training
* Run `bash main.sh $ --train` for Multi-GPU training, `$` is number of GPUs
### Test
* Configure your dataset path in `main.py` for testing
* Run `python main.py --test` for testing
### Demo
* Configure your image path in `main.py` for visualizing the demo
* Run `python main.py --demo` for demo
### Results
