Skip to content

A codebase & model zoo for pretrained backbone based on MegEngine.

License

Notifications You must be signed in to change notification settings

MegEngine/basecls

This branch is 2 commits behind megvii-research/basecls:main.

Folders and files

NameName
Last commit message
Last commit date
Jan 14, 2022
Jan 14, 2022
Apr 1, 2022
Jan 14, 2022
Jan 14, 2022
Sep 28, 2021
Sep 28, 2021
Jan 14, 2022
Jan 14, 2022
Sep 28, 2021
Apr 2, 2022
Jan 14, 2022
Jan 14, 2022
Sep 28, 2021
Apr 2, 2022
Jan 14, 2022
Jan 14, 2022
Jan 14, 2022
Jan 14, 2022

Repository files navigation

BaseCls

Documentation Status CI codecov

BaseCls 是一个基于 MegEngine 的预训练模型库,帮助大家挑选或训练出更适合自己科研或者业务的模型结构。

文档地址:https://basecls.readthedocs.io

安装

安装环境

BaseCls 需要 Python >= 3.6。

BaseCls 依赖 MegEngine >= 1.6.0。

通过包管理器安装

通过 pip 包管理器安装 BaseCls 的命令如下:

pip3 install basecls --user

默认不会安装包括 MegEngine 在内的部分依赖,可以通过以下命令进行完整安装:

pip3 install basecls[all] --user

对于 conda 用户, 可以选择通过在环境中先安装 pip,再按照上述方式进行 BaseCls 的安装。

通过源代码安装

为保证模型性能的可追溯性,避免实验碎片化,建议通过包管理器安装。如果包管理器安装的方式无法满足你的需求,则可以尝试自行通过源码安装。

安装依赖

pip3 install -r requirements.txt --user

安装 BaseCls

python3 setup.py develop --user

验证安装

在 Python 中导入 BaseCls 验证安装成功并查看安装版本:

import basecls
print(basecls.__version__)

开发者须知

开发环境

# 安装依赖
pip3 install -r requirements-dev.txt --user

# 配置 pre-commit
pre-commit install

开发流程

提交者需补充相应修改的单元测试。

# (外部开发者)fork repo,或(内部开发者)建立 new-feature 分支
git checkout -b new-feature

# 进行修改

# 代码风格检查与格式化
make lint
make format

# 单元测试与覆盖率检查
make unittest

# 提交修改
git commit

# 提交MR/PR

About

A codebase & model zoo for pretrained backbone based on MegEngine.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.9%
  • Makefile 0.1%