Skip to content

目前测试语义分割模型在kaggle无人机数据集上的效果,仅用于小组内学习与讨论。

Notifications You must be signed in to change notification settings

ai-bug-group/Aerial-Semantic-Segmentation-Drone

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

33 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Aerial-Semantic-Segmentation-Drone

Multi-GPU:

       Example:
       config:{'strategy': 
       tf.distribute.MirroredStrategy(devices=[
                                               "/gpu:0", 
                                               "/gpu:1", 
                                               "/gpu:2"])
       CUDA_VISIBLE_DEVICES="2,3,4" python train_v3.py

Single-GPU:

       Example:
       CUDA_VISIBLE_DEVICES="2" python train.py 

TODO:

       deeplabv3的tf.estimator下多GPU训练(改为MirroredStrategy)√
       复现的deeplabv3+在料口对齐上的效果 mIOU=0.9983 √
       MASK-RCNN源码(instance-segmentation)√
       增加图片强化:缩放、crop等
       预测、评估的代码

About

目前测试语义分割模型在kaggle无人机数据集上的效果,仅用于小组内学习与讨论。

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages