apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
2,329 Commits 1 Branch 0 Tags
26fc4fb01840440e5326862e3a236f3893d056c6
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 26fc4fb018 dataloader default color to imagenet mean 114 2020-04-05 18:02:41 -07:00
Glenn Jocher 2baf4e3f93 imagenet normalization on layer 0 batchnorm2d() 2020-04-05 17:33:06 -07:00
Glenn Jocher b70cfa9a29 mAP updates for rect inference 64 commit 2020-04-05 17:18:24 -07:00
Glenn Jocher c6d4e80335 move inference augmentation to model.forward() 2020-04-05 17:14:26 -07:00
Glenn Jocher 4da5c6c114 rect padding to 64, mAP increase 42.7 to 42.9 2020-04-05 16:06:27 -07:00
Glenn Jocher bb59ffe68f model forward() zip() removal 2020-04-05 15:22:32 -07:00
Glenn Jocher a657345b45 add FeatureConcat() module 2020-04-05 14:47:41 -07:00
Glenn Jocher 968b2ec004 .fuse() after .eval() 2020-04-05 14:05:12 -07:00
Glenn Jocher d04738a27c forward updated if-else 2020-04-05 13:49:13 -07:00
Glenn Jocher e81a152a92 tensorboard notice and model verbose option 2020-04-05 13:35:58 -07:00
Glenn Jocher a19b1a3b94 line thickness 2020-04-05 11:10:05 -07:00
Glenn Jocher 6203340888 detect.py multi_label default False 2020-04-05 11:05:49 -07:00
Glenn Jocher 41246aa042 tensorboard updates 2020-04-04 19:34:39 -07:00
Glenn Jocher 00c1fdd805 add MixConv2d() layer 2020-04-03 20:03:44 -07:00
Glenn Jocher eb9fb245aa add support for standalone BatchNorm2d() 2020-04-03 14:21:47 -07:00
Glenn Jocher 682c2b27e7 smart bias bug fix 2020-04-03 12:42:09 -07:00
Glenn Jocher 41a002e798 grid.float() 2020-04-03 12:38:08 -07:00
Glenn Jocher 93055a9d58 create_grids() to YOLOLayer method 2020-04-02 20:23:55 -07:00
Glenn Jocher 91f563c2a2 create_grids() to YOLOLayer method 2020-04-02 19:10:51 -07:00
Glenn Jocher 207c6fcff9 merge NMS full matrix 2020-04-02 18:53:40 -07:00
Glenn Jocher aa4591d7e9 batchnorm momentum to 0.03 2020-04-02 14:10:45 -07:00
Glenn Jocher 9155ef3f4f burnin merged with prebias 2020-04-02 14:08:21 -07:00
Glenn Jocher 27c7334e81 new layers.py file 2020-04-02 12:22:15 -07:00
Glenn Jocher 4ac60018f6 FLOPS report 2020-04-01 14:05:41 -07:00
Glenn Jocher ea80ba65af documentation updates 2020-04-01 12:16:37 -07:00
Glenn Jocher 765b8f3a3b documentation update 2020-04-01 12:12:14 -07:00
Glenn Jocher 5b322b6038 nvcr update 2020-04-01 09:57:00 -07:00
Glenn Jocher 300e9a7ad6 merge NMS full matrix 2020-03-31 21:31:09 -07:00
Glenn Jocher 8d788e10c4 mAP updates 2020-03-31 19:07:41 -07:00
Glenn Jocher 02802e67f2 merge NMS full matrix 2020-03-31 18:18:08 -07:00
Glenn Jocher f4eecef700 merge NMS speed/memory improvements 2020-03-31 15:37:23 -07:00
Glenn Jocher 992e0d7cb4 default test --conf to 0.001 2020-03-31 14:36:25 -07:00
Glenn Jocher 98271eb6ed remove deprecated models 2020-03-31 14:27:10 -07:00
Glenn Jocher 16862ea846 update 'reproduce our results' 2020-03-30 21:21:45 -07:00
Glenn Jocher b2d9f1898f burnin lr ramp 300 iterations 2020-03-30 19:27:42 -07:00
Glenn Jocher ac2aa56e0a feature fusion update 2020-03-30 17:53:17 -07:00
Glenn Jocher 108334db29 FLOPs update 2020-03-30 16:04:08 -07:00
Glenn Jocher 105882b3c6 GFLOPs correction 2020-03-30 15:30:53 -07:00
Glenn Jocher de52a008a5 default --img-size to 512 2020-03-30 11:46:20 -07:00
Glenn Jocher f6fc9634ab mAP updates 2020-03-30 11:37:38 -07:00
Glenn Jocher eb151a881e NMS and test batch_size updates 2020-03-29 20:41:32 -07:00
Glenn Jocher c6b59a0e8a LR schedule to 0.05 min 2020-03-29 13:29:06 -07:00
Glenn Jocher 9c5e76b93d EMA implemented by default 2020-03-29 13:14:54 -07:00
Glenn Jocher dc8e56b9f3 mAP update 2020-03-28 16:03:46 -07:00
Glenn Jocher ce17c26759 mAP updates 2020-03-27 13:52:07 -07:00
Glenn Jocher f9d34587da Merge NMS update 2020-03-27 13:11:24 -07:00
Glenn Jocher 4a63b24b09 Merge NMS update 2020-03-26 19:50:29 -07:00
Glenn Jocher 01ee0c5e95 Merge NMS update 2020-03-26 18:41:04 -07:00
Glenn Jocher dad59220f1 speed and comments update 2020-03-26 18:34:20 -07:00
Glenn Jocher faab52913c mAP updates 2020-03-26 16:35:46 -07:00
Glenn Jocher 9568d4562d mAP updates 2020-03-26 16:22:58 -07:00
Glenn Jocher 5ab13e5aa2 Merge remote-tracking branch 'origin/master' 2020-03-26 16:20:11 -07:00
Glenn Jocher 470371ba59 Test augment update 2020-03-26 16:20:06 -07:00
Glenn Jocher a4721e90f8 defult batch-size to 16 2020-03-26 14:22:59 -07:00
Glenn Jocher 4a7d9bdba9 mAP increases 2020-03-26 14:14:52 -07:00
Glenn Jocher a322fc5d4b Merge NMS update 2020-03-26 12:48:00 -07:00
Glenn Jocher 171b4129b5 Merge NMS update 2020-03-26 12:33:12 -07:00
Glenn Jocher eac07f9da3 Merge NMS update 2020-03-26 12:20:01 -07:00
Glenn Jocher 94344f5bea test augmentation comments 2020-03-26 11:34:32 -07:00
Glenn Jocher f91b1fb13a merge_batch NMS method 2020-03-26 11:28:46 -07:00
Glenn Jocher c71ab7d506 augmented testing 2020-03-26 11:25:44 -07:00
Glenn Jocher 23b34f4db8 merge_batch NMS method 2020-03-25 23:29:33 -07:00
Glenn Jocher aa0c64b5ac merge_batch NMS method 2020-03-25 23:24:57 -07:00
Glenn Jocher 3265d50f69 speed update 2020-03-19 18:15:09 -07:00
Glenn Jocher 89b6377723 Fuse by default when test.py called directly (faster) 2020-03-19 18:11:08 -07:00
Glenn Jocher fff45c39a8 cleanup/speedup 2020-03-19 16:41:42 -07:00
Glenn Jocher 1b68fe7fde cleanup 2020-03-19 16:23:44 -07:00
Glenn Jocher 83c9cfb7de FocalLoss() and obj loss speed and stability update 2020-03-19 12:30:37 -07:00
Glenn Jocher 20454990ce FLOPS report 2020-03-19 12:30:07 -07:00
Glenn Jocher 60c8d194cd FocalLoss() and obj loss speed and stability update 2020-03-19 11:52:52 -07:00
Glenn Jocher b3adc896f9 focal and obj loss speed/stability 2020-03-16 21:40:57 -07:00
Glenn Jocher 448c4a6e1f Remove deprecated --arc architecture options, implement --arc default for all cases 2020-03-16 20:46:25 -07:00
Glenn Jocher 77c6c01970 EMA class updates 2020-03-16 17:51:40 -07:00
Glenn Jocher 1a12667ce1 loss function cleanup 2020-03-16 17:31:37 -07:00
Glenn Jocher f1208f784e updated run history 2020-03-16 15:36:56 -07:00
Glenn Jocher 2a12a91245 nvcr.io/nvidia/pytorch:20.02-py3 2020-03-16 15:19:58 -07:00
Glenn Jocher c09fcfc4fe EMA class updates 2020-03-16 14:18:56 -07:00
Glenn Jocher c4047000fe FocalLoss() updated to match TF 2020-03-16 14:03:50 -07:00
Glenn Jocher 07d2f0ad03 test/inference time augmentation 2020-03-15 18:39:54 -07:00
Glenn Jocher adba66c3a6 EMA class updates 2020-03-14 18:08:48 -07:00
Glenn Jocher 851c9b9883 EMA class updates 2020-03-14 17:34:13 -07:00
Glenn Jocher d91469a516 EMA class updates 2020-03-14 17:33:29 -07:00
Glenn Jocher 5ebbb2db28 ASFF implementation 2020-03-14 17:04:38 -07:00
Glenn Jocher ea4c26b32d BatchNorm2d() to EfficientDet defaults: decay=0.997 eps=1e-4 2020-03-14 16:54:04 -07:00
Glenn Jocher 9ce4ec48a7 model.info() method implemented 2020-03-14 16:46:54 -07:00
Glenn Jocher b89cc396af EMA class updates 2020-03-14 16:23:14 -07:00
Glenn Jocher a52c0abf8d updates 2020-03-13 20:12:54 -07:00
Glenn Jocher 418269d739 FocalLoss() gamma and alpha default values 2020-03-13 16:51:30 -07:00
Glenn Jocher 208b9a73fe updates 2020-03-13 16:08:49 -07:00
Glenn Jocher f30a8706e5 update to coco results82 2020-03-13 12:06:48 -07:00
Glenn Jocher 0de07da612 updates 2020-03-13 11:03:39 -07:00
Glenn Jocher 6aae5aca64 inplace clip_coords() clamp 2020-03-13 10:47:00 -07:00
Glenn Jocher 731305142b json dict bug fixes and speed improvements 2020-03-13 10:35:58 -07:00
Glenn Jocher 5362e8254e updates 2020-03-13 10:20:52 -07:00
Glenn Jocher a2f5bc477c Merge remote-tracking branch 'origin/master' 2020-03-13 10:06:22 -07:00
Glenn Jocher a278623067 replaced floatn() with round() 2020-03-13 10:06:17 -07:00
Glenn Jocher 32404890e9 Update greetings.yml 2020-03-12 13:35:21 -07:00
Glenn Jocher 1ca2b8712a Update greetings.yml 2020-03-12 13:34:27 -07:00
Glenn Jocher 8a1f35eac6 updates 2020-03-12 01:09:17 -07:00
Glenn Jocher 41bf46a419 updates 2020-03-11 22:11:19 -07:00