基于上下文的停车位检测
(IEEE Access)的Tensorflow实现
此实现基于
准备数据集(PIL暂存区)
此代码一开始只能运行一次。
下载火车数据集
解压缩到$ your_data_path / train文件夹
下载测试数据集
解压缩到$ your_data_path / test文件夹
数据扩充,创建tfrecord和文本文件
python prepare_data.py --data_path = $ your_data_path
训练数据集
下载预训练重量(更新2020.10.26)
保存到“基于上下文的检测”文件夹下的“ pre_weight”文件夹
python train.py --data_path = $ your_data_path
训练有素的体重路径
停车上下文识别器的权重文件保存到“ weight_pcr / YYYYMMDD_HHMM”
针对并
评论0
最新资源