search

Reagan1311 / DABNet

Depth-wise Asymmetric Bottleneck for Real-time Semantic Segmentation (BMVC2019)
https://github.com/Reagan1311/DABNet
MIT License
140 stars 36 forks source link

custom data trainning #10

Closed zhangyunming closed 5 years ago

zhangyunming commented 5 years ago

hi，你好，我想咨询下两个问题：

  1. 如果我只想用指定的一个卡，是不是可以直接在train.py中改为1（我想用第2张卡）
  2. 如果我训练的数据只有两类，怎么修改cityscapes.pkl中的内容，如class_weights等

谢谢啊，老铁

Reagan1311 commented 5 years ago

1.是的 2.这个要修改cityscapes.py中的cityscapesTrainInform函数，需要重新计算class_weights