NielsRogge/transformers

Pull Requests Commits

Add first draft

NielsRogge committed 3 years ago

6ce35df2

More improvements

NielsRogge committed 3 years ago

5fd13f7c

Switch to channels first for img_metas

NielsRogge committed 3 years ago

a2c56d2e

Remove multi_apply

NielsRogge committed 3 years ago

e629c23c

Update README, remove multi_apply

NielsRogge committed 3 years ago

0cd9ea57

Add print statements

NielsRogge committed 3 years ago

127a47a8

Comment out most print statements

NielsRogge committed 3 years ago

3b68f08b

Simplify RPN forward

NielsRogge committed 3 years ago

d97dfe1b

NielsRogge committed 3 years ago

949fdbde

Update integration test

NielsRogge committed 3 years ago

c37d9a06

Fix docstrings, remove multi_apply, update integration test

NielsRogge committed 3 years ago

c76821ec

Apply some more suggestions

NielsRogge committed 3 years ago

edcdc98c

Apply suggestions from code review

NielsRogge committed 3 years ago

9091bd88

Fix training integration test

NielsRogge committed 3 years ago

bb703125

Delete scripts and add integration test for training

NielsRogge committed 3 years ago

787cbdac

Add integration test, general fixes

NielsRogge committed 3 years ago

afec3ae5

Remove img_metas default value

NielsRogge committed 3 years ago

c5772c1a

Activate mask loss

NielsRogge committed 3 years ago

447706fa

Deactive mask loss

NielsRogge committed 3 years ago

e48a9f56

NielsRogge committed 3 years ago

ef72d7bc

NielsRogge committed 3 years ago

87d9002c

Add losses to output

NielsRogge committed 3 years ago

ebd6d8c3

Improve docstrings

NielsRogge committed 3 years ago

9aa6f66d

Add loss computation of mask head

NielsRogge committed 3 years ago

419bb7a0

Add loss calculation of mask head

NielsRogge committed 3 years ago

775980c7

Add forward_train to roi head

NielsRogge committed 3 years ago

5048d212

More improvements

NielsRogge committed 3 years ago

a2c25c15

Add training support for RPN

NielsRogge committed 3 years ago

6727b0e5

Improve forward of RPN

NielsRogge committed 3 years ago

1be2472a

NielsRogge committed 3 years ago

76dbb5e9

Older