OpenMMLab Detection Toolbox and Benchmark
Project description
Introduction
This is pretrained models for person search. (resnet50,resnet18,shufflenetv1)
License
This project is released under the Apache 2.0 license.
Installation
This project is developed upon MMdetection, please refer to install.md to install MMdetection.
We utilized mmcv=1.3.9, pytorch=1.7.0
pip install psvis
Usage
- resnet50
from psvis.models.backbones import resnet
resnet50 = resnet.__dict__['resnet50'](pretrained=True)
- resnet18
from psvis.models.backbones import resnet
resnet18 = resnet.__dict__['resnet18'](pretrained=True)
- shufflenetv1
from psvis.models.backbones import shufflenet_v1
shufflenetv1 = shufflenet_v1.__dict__['shufflenetv1'](pretrained=True)
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
psvis-1.6.0.tar.gz
(25.5 kB
view hashes)
