NielsRogge/transformers

Pull Requests Commits
Fix img_metas
NielsRogge committed 2 years ago
cf9ff49c
Deactivate mask loss
NielsRogge committed 2 years ago
8cad1083
Activate mask loss
NielsRogge committed 2 years ago
0f0ca4f6
Deactive mask loss
NielsRogge committed 2 years ago
5fe9a6f9
Remove checké
NielsRogge committed 2 years ago
3367d106
Add cpu method
NielsRogge committed 2 years ago
3258c316
Add losses to output
NielsRogge committed 2 years ago
ccc44631
Improve docstrings
NielsRogge committed 2 years ago
498267b9
Add loss computation of mask head
NielsRogge committed 2 years ago
69ef89dd
Add loss calculation of mask head
NielsRogge committed 2 years ago
64e79f75
Add forward_train to roi head
NielsRogge committed 2 years ago
81222c68
More improvements
NielsRogge committed 2 years ago
8e80fb9d
Add training support for RPN
NielsRogge committed 2 years ago
e9be76b7
Improve forward of RPN
NielsRogge committed 2 years ago
f5ed75b8
Add sampler
NielsRogge committed 2 years ago
a917ccc7
Add Assigner and BboxOverlaps classes
NielsRogge committed 2 years ago
65e65b6f
Verify mask head
NielsRogge committed 2 years ago
8c66382a
Add mask head
NielsRogge committed 2 years ago
09491722
Make bounding box results match original implementation
NielsRogge committed 2 years ago
662510f0
Add initial draft of RoI head
NielsRogge committed 2 years ago
4bafe80e
Use torchvision's nms implementation
NielsRogge committed 2 years ago
0255a844
More improvements
NielsRogge committed 2 years ago
5b60a262
More improvements
NielsRogge committed 2 years ago
d3a6609e
More fixes
NielsRogge committed 2 years ago
459c3bc5
More fixes
NielsRogge committed 2 years ago
f8ccac39
Fix conversion without layernorm
NielsRogge committed 2 years ago
e151e543
More improvements
NielsRogge committed 2 years ago
b06b60dc
More progress
NielsRogge committed 2 years ago
9a42b463
Fix conversion script for backbone
NielsRogge committed 2 years ago
9ec94048
First draft
NielsRogge committed 2 years ago
6683796b
Terms Of Use Privacy Policy Refund Policy Impressum