用于构建深度学习分类程序的工具包。
wk-classif的Python项目详细描述
工作分类
用于构建深度学习分类程序的工具包。使用方便,重量轻,功能强大。在
安装
pip3 install wk-classify
使用
快速启动
^{pr2}$一个真实的例子
fromwcfimporttrain,TrainValConfigBase,val,t,EasyTransform,models_namesclassConfig(TrainValConfigBase):MODEL_TYPE=models_names.shufflenet_v2_x0_5TAG='[%s]'%(MODEL_TYPE)GEN_CLASSES_FILE=TrueUSE_tqdm_TRAIN=TrueINPUT_SIZE=(252,196)#(w,h)BATCH_SIZE=64MAX_EPOCHS=200BALANCE_CLASSES=TrueVAL_INTERVAL=1WEIGHTS_SAVE_INTERVAL=1WEIGHTS_INIT='weights/training/model_best.pkl'TRAIN_DIR='/home/ars/sda5/data/projects/烟盒/data/现场采集好坏烟照片/相机1-train'VAL_DIR='/home/ars/sda5/data/projects/烟盒/data/现场采集好坏烟照片/相机1-val'val_transform=EasyTransform([t.Resize(INPUT_SIZE[::-1]),t.SaveToDir('data/test'),t.ToTensor(),])train_transform=EasyTransform([t.ColorJitter(brightness=0.2,contrast=0,saturation=0,hue=0),# t.RandomHorizontalFlip(),# t.RandomVerticalFlip(),# t.RandomRotate(360),t.RandomTranslate(30),t.RandomBlur(p=0.3,radius=1),t.RandomSPNoise(p=0.3),*val_transform,])# def get_model(self, num_classes=None):# model=YourModel(...)# return modelif__name__=='__main__':cfg=Config()train(cfg)# res=val(cfg)# print(res)
所有选项
有关所有选项,请参见TrainValConfigBase
类
如何预测?
请参见demo_predict.py
更多
请参见demo_train.py
和{
- 项目
标签: