-
Unofficial implementations of some classical backbone CNNs with pytorch
-
This is pretty much just a practising project. If you want to do some research, I strongly recommend using officially issued models in torchvision.
-
Requirements:
· torch,torch-vision
· torchsummary
· mlflow
· onnx onnnruntime
· openvino
-
A training script is supplied in “train.py”,the arguments are in “utils/args.yml".
-
new features:
- mlflow traning track. You need a mlflow server, and to intall mlflow on you training server.
- onnx inference.
- openvino inference.
N.B. If you don't want these features, pelease check release branch v1.0.0.
-
Notifications
You must be signed in to change notification settings - Fork 43
YichaoCai1/torch_backbones
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Unofficial implementations of some classical CNN backbones with pytorch
Topics
Resources
Stars
Watchers
Forks
Packages 0
No packages published