Skip to content

lauraset/SAM-UTC-WSSS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SAM-UTC-WSSS

Paper

Will be available soon

Setup environments

torch              1.13.1+cu117
torchvision        0.14.1+cu117
tree-filter               0.1

Install tree-filter on Windows (or Linux)

cd treefilter/lib_tree_filter
revise cuda_lib_path in setup.py
python setup.py build
python setup.py install 

Setup data

set the datapath in allconfig.py

DATA_PATH_HOME = '/localssd/yinxia/data'  # store data
SAVE_WEIGHT_PATH = '/localssd/yinxia/CRGNetdata/' # store weights

download pretrained weights

See Release or download from baiduwangpan: https://pan.baidu.com/s/1o5AnqWEULwTLNBIM-wpwEg, code: h7dc

Training

python allrun_zurich_Transform3.py