Repository files navigation
MegEngine_CU11 : 包含 CUDA11 支持的 MegEngine Python Wheel 包
MgeEditing :基于 MegEngine 的图像库
Echo :一个优秀的算子库
ClearML(Allegro Trains) : ML/DL 开发和生产套件,包含 Experiment Manager / ML-Ops / Data-Management 三大核心模块
MegEngine.js :MegEngine javascript 版本,可以在 javascript 环境中快速部署 MegEngine 模型
commitlint config megengine :MegEngine commitlint 配置项
Cascade RCNN :一种 Cascade RCNN 实现
Megvision :一些经典 CV 模型的实现和权重
Models :MegEngine 实例训练代码,包含各类任务的基本实现,是初学者最好的参考物
MgeConvert :MegEngine Traced module 模型格式转换器,支持 ONNX / TFLite / Caffe 等各类导出格式
MegFile :一个可以完美抽象 S3、HTTP、本地文件等协议的 python 文件接口库,是 smart-open 库的升级版
MegFlow :面向计算机视觉应用的流式计算框架,提供了一套可快速完成 AI 应用部署的视觉解析服务方案
MegSpot :一款提供免费免登录、高效、专业、跨平台的图片&视频的对比的 PC 应用工具
BaseCls :极其全面的分类模型库,提供了海量模型的训练代码和预训练权重,全部算法均可以快速部署到硬件上
Swin Transformer :Swin Transformer 的实现,支持 DTR 功能减少显存占用量
MegFLow-Cat_Feeder :基于 MegFlow 框架的猫咪检测以及自动投喂的解决方案
MegPeak :处理器性能测试工具
Netron 已可视化 MegEngine TracedModule,欢迎使用 模型示例文件 体验
MegCC :一个运行时超轻量,高效,移植简单的深度学习模型编译器
BaseDet :提供了一些经典的检测 SOTA 模型以及相关组件
mperf :一个算子性能调优工具箱
MegEngine 原创论文复现:
RepLKNet :[CVPR 2022] Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs
GyroFlow : [ICCV 2021] GyroFlow: Gyroscope-Guided Unsupervised Optical Flow Learning
ICD : [NeurIPS 2021] Instance-Conditional Knowledge Distillation for Object Detection
FINet : [AAAI 2022] FINet: Dual Branches Feature Interaction for Partial-to-Partial Point Cloud Registration
OMNet : [ICCV 2021] OMNet: Learning Overlapping Mask for Partial-to-Partial Point Cloud Registration
CREStereo : [CVPR 2022] Practical Stereo Matching via Cascaded Recurrent Network with Adaptive Correlation
D2C-SR : [ECCV 2022] D2C-SR: A Divergence to Convergence Approach for Real-World Image Super-Resolution
FST-Matching : [ECCV 2022] Explaining Deepfake Detection by Analysing Image Matching
zipfls : [ECCV 2022] Zipf's LS: Efficient One Pass Self-distillation with Zipf's Label Smoothing
HDR-Transformer : [ECCV 2022] Ghost-free High Dynamic Range Imaging with Context-aware Transformer
经典论文模型对应结构的 MegEngine inference 函数:
You can’t perform that action at this time.