深度学习之PyTorch物体检测实战
说明
- 这是《深度学习之PyTorch物体检测实战》这本书对应的代码,书籍将会在2019年年底由机械工业出版社出版。
- 物体检测是一个十分注重实践的知识点,因此强烈建议读者下载此代码,结合书籍进行实践学习。
- 由于工程量较大,因此本书的Faster RCNN、SSD与YOLO三大网络的实现皆借鉴了其他作者的实现,并增加了一些注解、尽量保留最小实现模块,读者也可以根据自身需求来选择。
- 在运行时遇到任何问题,欢迎在此提交issue,或者到其他repo的issue里寻找答案。
- Enjoy Coding!
环境版本
- PyTorch:0.4.0
- Python:3.6
- CUDA:9.0
目录
- chapter1:浅谈物体检测与PyTorch
- chapter2:PyTorch基础
- chapter3:网络骨架-Backbone
- chapter4:两阶经典检测器:Faster RCNN
- chapter5:单阶多层检测器:SSD
- chapter6:单阶经典检测器:YOLO
- chapter7:模型加速之轻量化网络
- chapter8:物体检测细节处理
- chapter9:物体检测难点
- chapter10:物体检测的未来发展