在TensorFlow 2.0中实现的YoloV3
此仓库使用所有最佳实践在TensorFlow 2.0中提供了YoloV3的干净实现。
主要特点
TensorFlow 2.0
yolov3具有预先训练的权重
yolov3-tiny具有预先训练的权重
推论实例
转移学习的例子
使用tf.GradientTape急切模式训练
使用model.fit图模式训练
具有tf.keras.layers功能模型
使用tf.data输入管道
Tensorflow服务
向量化转换
GPU加速
完全集成的absl-py从
干净的实施
遵循最佳做法
麻省理工学院执照
用法
安装
conda(推荐)
# Tensorflow CPU
conda env create -f conda-cpu.yml
conda activate yolov3-tf2-cpu
# Tensorflow GPU
conda env create -f conda-gpu.yml
conda activate yolov3-tf2-gpu
点子
pip install -r requireme
评论0
最新资源