python_developer_tools
python_developer_tools copied to clipboard
论文复现,多机多卡
python_developer_tools
python 开发过程中常用到的工具;包括网站开发,人工智能,文件,数据类型转换 支付接口对接,外挂,bat,sh等我在工作中常用到的开发工具
- 制作一个可插拔的python开发工具
- 论文复现
- 深度学习tricks收集
所有的论文代码都在这里
安装和使用
pip uninstall python_developer_tools
pip install git+https://github.com/carlsummer/python_developer_tools.git
from python_developer_tools import cv
Contents
-
cv 计算机视觉
-
基础组成成分
-
input_conv 输入给网络之前的一步操作
- SpaceToDepth
-
Convolution Series
- Depthwise Separable Convolution Usage
- DepthSpatialSepConv
- GroupConv
- SpatialSepConvSF
- DepthConv
- MBConv
- Involution
- DY_Conv2d
- CoordConv2d
-
attentions注意力机制
- Squeeze-and-Excitation Networks
- Residual Attention
-
BN
- inplace_abn
-
activates 激活函数
- ReLU
- GELU
- LeakyReLU
- relu6
- SiLU
- sigmoid
- Swish
- Adaptively-Parametric-ReLU
- DynamicReLU
-
softmax序列
- softmax
- norm_softmax
- sinmax
- norm_sinmax
- cosmax
- norm_cosmax
- sin_2_max
- norm_sin_2_max
- sin_2_max_move
- norm_sin_2_max_move
- sirenmax
- norm_sirenmax
- sin_softmax
- norm_sin_softmax
- serf
- [全连接FC]
- SSM
- SwishLinear
-
池化pool
- SoftPool
- adaPool
- empool
- eidwpool
- idwpool
- GlobalAvgPool2d
- SwishAdaptiveAvgPool2d
- BlurPool
- SPP
- MaxGroupPooling
- regularization
-
input_conv 输入给网络之前的一步操作
-
分类classes
- demo
- 训练cifar10数据集
- demo
- datasets
-
数据增强
- 分类任务数据增强
- 图片自动对比度
- 直方图增强
- mix_data
-
APR
- posterize
- rotate
- solarize
- shear_x
- shear_y
- translate_x
- translate_y
- color
- contrast
- brightness
- sharpness
- 直线检测数据增强
- cutout
- 旋转透视变换
- 分类任务数据增强
- 数据集读取
-
数据增强
- [detection]
-
YOLOX
- YOLOX训练自己的数据集
-
YOLOX
-
scheduler
- ExpLR
- WarmupExponentialLR
- StepLR
- WarmupStepLR
- MultiStepLR
- WarmupMultiStepLR
- CosineLR
- WarmupCosineAnnealingLR
- LambdaLR
- ReduceLROnPlateau
- CosineAnnealingWarmRestarts
- CyclicLR
- OneCycleLR
- PolyLR
-
optimizer
- SGD
- ASGD
- Adagrad
- Adadelta
- RMSprop
- Adam
- Adamax
- SparseAdam
- L-BFGS
- Rprop
- AdamW
- RAdam
- Ranger
- LARS
-
post_processing后处理
- tta
- [pre_processing]
- learnableResizer
-
loss
- classes
- CrossEntropyLoss
- focalloss
- LabelSmoothingCrossEntropy
- TripletLoss
- detection
- lines
- CenterLoss
- Diceloss
- focalloss
- OHEMloss
- OIMloss
- topk_crossEntrophy
- NLLLoss
- classes
- train
-
不同数量卡训练
- 单机单卡
- 单机多卡
- 多机多卡
- 二阶段训练
- swa
- swaoffline
- 自动混合精度训练
-
对抗训练
- FGSM
- BIM
- RFGSM
- CW
- PGD
- PGDL2
- EOTPGD
- FFGSM
- TPGD
- MIFGSM
- VANILA
- GN
- APGD
- APGDT
- FAB
- Square
- AutoAttack
- OnePixel
- DeepFool
- DIFGSM
-
不同数量卡训练
- utils
- tensorboard
- torch_utils
- label标准工具
-
基础组成成分
-
files
- common
- pickle
-
machinelearning
- KFold
- SVM
- knn
- [python]
-
threadings
- multiprocessing_utils
-
threadings
Convolution-series
名称 | csdn |
---|---|
Depthwise Separable Convolution | |
MBConv | |
Involution |
activates
名称 | csdn |
---|---|
Adaptively-Parametric-ReLU | |
DynamicReLU | 解析 |
torch_utils
名称 | 功能 |
---|---|
init_seeds | 固定pytorch训练时所有的随机种子 |
cuda2cpu | 将pytorch的变量从cuda内存中移动到cpu的内存中 |
select_device | 选择训练设备 |
collect_env_info | 打印python,cuda,cudnn,torch等版本 |
labels_to_image_weights | 根据图片样本数量计算weights |
init_cudnn | 加速训练还是追求模型性能 |
get_world_size | 返回全局的整个的进程数 |
get_rank | 执行该脚本的进程的rank |
get_model_info | 计算模型的参数量和计算一张图片的计算量 |
files
common
名称 | 功能 |
---|---|
get_filename_suf_pix | 获取路径的文件名,后缀,父路径 |
pickle
名称 | 功能 |
---|---|
write_pkl | 将数据存储为pkl |
read_pkl | 读取pkl文件的内容 |