20200622:mobilenetv3-tensorflow实战(一):项目准备和代码调试

1、开源代码:

2、项目准备:

  • 项目克隆:git clone https://github.com/Bisonai/mobilenetv3-tensorflow
  • 环境搭建:
    • 创建虚拟环境:conda create -n "mobilenetv3_Anne" python=3.6
    • 环境激活:conda activate mobilenetv3_Anne 
    • gpu版tensorflow安装:pip --default-timeout=100 install tensorflow-gpu==1.15
    • scikit-image安装:pip --default-timeout=100 install scikit-image
    • tqdm安装:pip --default-timeout=100 install tqdm
    • tensorflow_datasets安装:pip install tensorflow_datasets
    • ipython安装:conda install ipyton
    • ipykernel安装:conda install ipykernel
    • 将环境写入jupyter核中:python -m ipykernel install --name moblienetv3_Anne

3、项目代码调试

由于不清楚作者的tensorflow版本,因此直接粗暴调试,报错的地方直接修改为tensorflow 1.15里的用法即可。

注意:这里一定要查官方文档,以官方文档为准,网上很多资料并没有说明用法到底是哪个版本,直接套用解决不了问题。

  • 修改dataset.py里build_dataset:
  • 修改train.py里modle.fit:

!!! 训练顺利跑起来了,放图:

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值