带有Tensorflow 2的YOLOv1
为了易于实现,我没有实现与纸张完全相同的实现。 下面介绍的内容与本文的实现方式有所不同。
骨干网。 (我使用Xception代替了本文中提到的网络。)
学习率表(我使用了tf.keras.optimizers.schedules.ExponentialDecay )
超级参数
数据扩充
等等 。 。 。
预览
即将更新。 。 。
使用Docker构建环境
构建Docker映像
$ docker build -t ${NAME} : ${TAG} .
创建一个容器
$ docker run -d -it --gpus all --shm-size= ${PROPER_VALUE} ${NAME} : ${TAG} /bin/bash
Pascal VOC数据集()
带有Pascal VOC数据集
图片数量
火车
验证
测试
帕斯卡VOC
1