QtacierP / BaselineZoo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Baseline Model Zoo

This project mainly focuses on the baseline model for classification/segmentation/object detection, including resnet, resnext, densenet; deeplab, unet; faster-rcnn, yolo...

This is an in-progress project, if you are interested in it, thanks for starring it:)

Features

Metrics

  • In-implement metrics

  • friendly metrics with auto-grad and auto-log

Classification

  • ResNet Family & ResNext Family

  • DenseNet Family

  • Learning Rate Schdule and automatical finder

Segmentation

  • DeepLab Family (Need test)

  • Image loggers (Some issueses)

About


Languages

Language:Python 100.0%