简体中文 | English
miemieGAN是咩酱基于YOLOX进行二次开发的个人检测库(使用的深度学习框架为pytorch), 支持单机单卡、单机多卡、多机多卡训练模式(多卡训练模式建议使用Linux系统),支持Windows、Linux系统,以咩酱的名字命名。miemieGAN是一个不需要安装的检测库用户可以直接更改其代码改变执行逻辑,所见即所得!所以往miemieGAN里加入新的算法是一件很容易的事情。得益于YOLOX的优秀架构,miemieGAN里的算法训练速度都非常快,数据读取不再是训练速度的瓶颈!目前miemieGAN支持StyleGAN2ADA、StyleGAN3等算法,预计未来会加入更多算法,所以请大家点个star吧!
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
torch版本建议1.10.1+cu102或者更高;torchvision版本建议0.11.2+cu102或者更高。
- 【2022/07/12】 StyleGAN3算法支持导出到NCNN!也支持生成视频!详情请参考README_StyleGAN 文档的“如何导出ncnn?”小节。
- 【2022/07/08】 StyleGAN2ADA算法支持导出到NCNN!详情请参考README_StyleGAN 文档的“如何导出ncnn?”小节。技术细节请参考本人的知乎文章全网第一个!stylegan2-ada和stylegan3的pytorch实现二合一!支持导出ncnn!
算法1群:645796480(人已满)
算法2群:894642886
粉丝群:704991252
关于仓库的疑问尽量在Issues上提,避免重复解答。
B站不定时女装: _糖蜜
知乎不定时谢邀、写文章: 咩咩2013
西瓜视频: 咩咩2013
微信:wer186259
本人微信公众号:miemie_2013
技术博客:https://blog.csdn.net/qq_27311165
AIStudio主页:asasasaaawws
欢迎在GitHub或上面的平台关注我(求粉)~
如果你觉得这个仓库对你很有帮助,可以给我打钱↓
咩酱爱你哟!