apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
2,380 Commits 1 Branch 0 Tags
76fb8d48d43da1a46fcbe1d7142a1947ef162da6
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 76fb8d48d4 ng dependence removed from build_targets() 2020-04-13 21:25:03 -07:00
Glenn Jocher 0dd5f8eee8 code cleanup 2020-04-13 18:25:59 -07:00
Glenn Jocher ca3a9fcb0b return get_yolo_layers() 2020-04-13 17:56:12 -07:00
Glenn Jocher b8574add37 new find_modules() fcn 2020-04-13 17:48:30 -07:00
Glenn Jocher 77e6bdd3c1 FLOPs at 480x640, BN init 2020-04-12 18:44:18 -07:00
Glenn Jocher 1038b0d269 multi-scale update 2020-04-12 18:22:54 -07:00
Glenn Jocher 46726dad13 torch.tensor(ng, device=device) 2020-04-12 13:02:00 -07:00
Glenn Jocher efc754a794 add generations arg to kmeans() 2020-04-12 12:49:23 -07:00
Glenn Jocher f65a50d13d Merge remote-tracking branch 'origin/master' 2020-04-12 10:20:28 -07:00
Glenn Jocher eda31c8bd0 print speeds for save_json 2020-04-12 10:20:21 -07:00
Glenn Jocher ed1d4f5ae7 k for kernel_size 2020-04-11 12:37:03 -07:00
Glenn Jocher a34219a54b pading from (k-1) // 2 to k // 2 2020-04-11 12:18:54 -07:00
Glenn Jocher b574f765ce add warning to plot_results() 2020-04-11 11:04:10 -07:00
Glenn Jocher 7be71b02e2 get_yolo_layers() 2020-04-11 10:56:20 -07:00
Glenn Jocher dcc2e99fb2 get_yolo_layers() 2020-04-11 10:55:49 -07:00
Glenn Jocher 58edfc4a84 kaiming weight init 2020-04-11 10:45:33 -07:00
Glenn Jocher 2cf23c4aee add MixConv2d() layer 2020-04-10 18:58:34 -07:00
Glenn Jocher 4bbfda5cde hist equalization 2020-04-10 17:29:57 -07:00
Glenn Jocher 9bc3a551d9 histogram equalization added to augmentation 2020-04-10 17:24:49 -07:00
Glenn Jocher 398f8eadec add thr=0.10 to kmean_anchors() 2020-04-10 16:34:32 -07:00
Glenn Jocher aa8b1098dd adapt mosaic to img channel count 2020-04-10 16:28:59 -07:00
Glenn Jocher 6736d7d125 swap cv2.INTER_AREA for cv2.INTER_LINEAR 2020-04-10 12:47:07 -07:00
Glenn Jocher b98ce11d3a add MixConv2d() layer 2020-04-09 20:20:23 -07:00
Glenn Jocher 6e19245dc8 auto strip optimizer from best.pt after training 2020-04-09 19:53:29 -07:00
Glenn Jocher bc74822540 notebook update 2020-04-09 14:33:24 -07:00
Glenn Jocher d1601ae0f3 training updates 2020-04-08 21:34:34 -07:00
Glenn Jocher 4120ac3aa6 training updates 2020-04-08 21:01:58 -07:00
Glenn Jocher c7f5b6cc21 speed update 2020-04-08 20:53:17 -07:00
Glenn Jocher d9af081b7f Merge remote-tracking branch 'origin/master' 2020-04-08 20:43:57 -07:00
Glenn Jocher 8f71b7865b run once to remove initial timing effects 2020-04-08 20:43:51 -07:00
Glenn Jocher 4f6843e249 Update README.md 2020-04-08 12:28:10 -07:00
Glenn Jocher 13fa2798a6 Update README.md 2020-04-08 12:27:33 -07:00
Glenn Jocher b6959a2f54 ONNX export self.training=False 2020-04-08 10:25:52 -07:00
Glenn Jocher 97780cfdb4 parameterize grid size 2020-04-08 10:14:33 -07:00
Glenn Jocher 9f41b7601a inference speed and mAP updates 2020-04-08 10:02:20 -07:00
Glenn Jocher f54d28ba63 improve assert no labels found 2020-04-08 09:57:59 -07:00
Glenn Jocher 933e05cb44 augment update 2020-04-07 17:35:35 -07:00
Glenn Jocher 6c5ecaf805 remove label loading during training 2020-04-07 16:57:22 -07:00
Glenn Jocher b3e1d74478 remove imwrite from augment 2020-04-07 14:23:31 -07:00
Glenn Jocher d79c3bd076 parameterize augment scales 2020-04-07 14:19:43 -07:00
Glenn Jocher b9b14bef59 scale_img() bug fix 2020-04-07 13:35:47 -07:00
Glenn Jocher 067ee264c0 scale_img() bug fix 2020-04-07 12:59:52 -07:00
Glenn Jocher 68f58f4dec scale_img() update 2020-04-07 12:51:52 -07:00
Glenn Jocher 1a3c77df95 Merge remote-tracking branch 'origin/master' 2020-04-07 12:27:56 -07:00
Glenn Jocher 1a511d2906 updated --augment sizes and results 2020-04-07 12:27:49 -07:00
Glenn Jocher 4fa3fd2df3 tensorboard notice 2020-04-06 16:33:23 -07:00
Glenn Jocher 05ae6e8499 tensorboard/focal loss reporting update 2020-04-06 15:45:18 -07:00
Glenn Jocher c7f93bae40 update coco-tuned hyp['cls'] to current dataset 2020-04-06 10:58:07 -07:00
Glenn Jocher 4fc0012829 initial batchnorm to 0.03 momentum 2020-04-05 18:03:49 -07:00
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