Skip to content

suojinhui/DA-Faster

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DA-Faster

环境配置

  1. 该项目基于detectron-v0.6,在使用前确保自己已经配置好了环境。
  2. 数据集注册在/da_faster/data/register中,使用COCO格式,需要使用者自行修改数据集路径。目前只提供cityscapes->foggy_cityscapes场景,其余场景自行注册。

预训练权重

在主目录下创建weights,然后从这里下载vgg-16的预训练权重,并放入weights文件夹中。

用法

  • da_traing:
python ./train_net_da.py --config-file ./configs/faster-rcnn-vgg16-CrossCityscapes.yaml --num-gpus 4
  • source_only_training:
python ./train_net_so.py --config-file ./configs/faster-rcnn-vgg16-CrossCityscapes_so.yaml --num-gpus 4

注意:想要更改训练配置,请在config/目录下更改训练config文件,或者在da_faster/config.py中更改。(cfg中的参数更改也支持命令行传参)

联系方式

如果有任何问题,欢迎联系我:[email protected]

About

基于detectron2-v0.6的DA-Faster

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages