add models - Githubissues

rai-project / pytorch

Other

6 stars 2 forks source link

add models #20

Open abhiutd opened 5 years ago

abhiutd commented 5 years ago

[x] torch vision (https://pytorch.org/docs/stable/torchvision/models.html)
[x] pytorch-ssd (https://github.com/qfgaohao/pytorch-ssd)
[ ] ssd.pytorch (https://github.com/amdegroot/ssd.pytorch#evaluation)
[ ] rai-project/pytorch_models (https://github.com/rai-project/pytorch_models)
[ ] torch script compatible branch of maskrcnn_Benchmark (https://github.com/t-vi/maskrcnn-benchmark/tree/scripting)
[ ] pytorch-1.0 branch of faster-rcnn.pytorch (https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0)

abhiutd commented 5 years ago

Image classification models

[x] cadene (https://github.com/Cadene/pretrained-models.pytorch)
[ ] bentrevett (https://github.com/bentrevett/pytorch-image-classification)
[ ] hysts (https://github.com/hysts/pytorch_image_classification)
[ ] warmspringwinds (https://github.com/warmspringwinds/pytorch-image-classification)