门控SCNN
基于ICCV 2019论文 SCNN PyTorch门控SCNN的实现,。
要求
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch
蒂姆
pip install timm
OpenCV
pip install opencv-python
城市景观
pip install cityscapesscripts
用法
火车模型
python train.py --epochs 175 --backbone_type resnet101
optional arguments:
--data_path Data path for cityscapes dataset [default value is 'data']
--backbone_type
2022-01-04 21:44:44
2.78MB
Python
1