Coder Social home page Coder Social logo

leeiadong / tianchi-2021-guangdong-tile-detection Goto Github PK

View Code? Open in Web Editor NEW

This project forked from zhengye1995/tianchi-2021-guangdong-tile-detection

0.0 0.0 0.0 6.52 MB

天池2021广东工业智能制造大赛瓷砖瑕疵检测极客奖方案

License: Apache License 2.0

Python 99.89% Shell 0.08% Dockerfile 0.03%

tianchi-2021-guangdong-tile-detection's Introduction

2021广东工业智造创新大赛 瓷砖瑕疵检测 解决方案

队伍：我好菜

比赛地址：2021广东工业智造创新大赛 瓷砖瑕疵检测

Siamese Attention FPN 特征差+spatial attention思路

core slides:

Aaron Swartz

  • Siamese Attention FPN
    • 基本框架：Cascade-RCNN
    • backbone： resnet50
    • neck: FPN 基础上引入特征差和空间attention
    • RPN:
      • anchor box assign 采用 ATSS
      • anchor_scale 随训练尺度一起缩小
    • cascade 三个head 根据比赛map计算iou进行对应调整
    • 采用fp16加速训练和增大输入面积来缓解部分小目标问题
  • 后处理
    • NMS
    • 最大score二类后处理, 根据单个图片bbox最高score来对图像进行二次过滤, 判断该图像是否是正常样本

代码环境及依赖 （阿里线上环境）

  • OS: Ubuntu
  • GPU: Telsa V100 (16GB) * 4 (实测训练只需要7-8g显存)
  • python: python3.7
  • nvidia 依赖:
    • cuda: 10或者11均可
    • cudnn: 7.5.1-8 均可
    • nvidia driver version: 430.14 - 最新均可
  • deeplearning 框架: pytorch1.6
  • 其他依赖请参考requirement.txt （和mmdetection一致）

模型训练及预测 （在比赛平台docker方式提交运行）

./run.sh

Contact

author：rill

email：[email protected]

tianchi-2021-guangdong-tile-detection's People

Contributors

zhengye1995 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.