用于目标检测的特征金字塔网络 注意 基于FPN的。 支持多GPU培训! 抽象 这是的的张量流重新实现。 该项目基于 ,由和完成。 对VOC 2007进行培训,对VOC 2007进行测试(PS。该项目还支持可可培训。) 比较方式 use_voc2007_metric 楷模 地图 羊 马 自行车 瓶子 牛 沙发 总线 狗 猫 人 培养 餐桌 飞机 汽车 盆栽 电视监视器 椅子 鸟 船 摩托车 resnet50_v1 73.09 72.11 85.63 77.74 55.82 81.19 67.34 82.44 85.66 87.34 77.49 79.13 62.6
1
图像分割 该项目在实现了用于语义分割的神经网络。 项目概况 该项目的主文件是convolutional_autoencoder.py ,其中包含用于数据集处理的代码(数据集类),模型定义(模型类)以及训练代码。 为了抽象模型中的图层,我们创建了layer.py类接口。 此类当前具有两个实现: conv2d.py和max_pool_2d.py 。 要推断训练后的模型,请查看infer.py文件。 最后,有几个文件夹: 数据*包含预处理的数据集(请注意,当前模型实现应至少与128x128图像一起使用。) imgaug包含用于数据扩充的代码( ) noteboks包含一些有趣的图像分割
1