语义分割:使用综合数据集的语义分割 源码

语义分割 这是一个使用从Unreal Engine获得的综合数据集进行语义分割的小项目。 任务是使用插件从Unreal Engine创建合成数据集并识别图像中的对象。 数据集由RGBD(RGB +深度)图像和地面真实情况组成。 环境很简单,房间里有沙发,两台电视,两棵植物和一些桌子。 一旦收集了数据集,就可以检测和定位图像中的对象。 这可以使用语义分割网络来完成。 语义分割网络基本上执行按像素分类,为每个像素分配其类别。 仓库描述 在语义段内,主要文件是: generate_dataset.py:用于生成与unrealcv服务器交互的数据集的脚本。 write_dataset_to_tf.