EfficientDet Pytorch+colab复现

最近zylo大神复现了pytorch版本的EfficientDet，趁着热度我对大神的github项目进行了测试

大神的GitHub链接

我们用google colab来进行测试
1.首先打开colab链接
2.文件=>新建笔记本
3.依次执行下面的代码块

配置环境

!pip install pycocotools numpy==1.16.0 opencv-python tqdm tensorboard tensorboardX pyyaml webcolors matplotlib
!pip install torch==1.4.0
!pip install torchvision==0.5.0

git大神的代码

!git clone https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch

切换到代码目录

import os
os.chdir( 'Yet-Another-EfficientDet-Pytorch')

创建权重目录

!mkdir weights

进入权重目录

os.chdir( 'weights')

下载pth

!wget https://github.com/zylo117/Yet-Another-Efficient-Pytorch/releases/download/1.0/efficientdet-d0.pth

返回上级目录

os.chdir('../')

在test文件夹中上传自己的图片
改变efficientdet_test.py中的img_path指向自己的图片
我上传了一张很多小黄车的图片
运行demo

!python efficientdet_test.py

运行结果

可以看到efficientdet D0的结果还是不错的

如果觉得有帮助记得点赞哦