apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
2,430 Commits 1 Branch 0 Tags
f799c156119bf9ea6cfed1fd201817606718e67e
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher f799c15611 result not updated from pycocotools 2020-04-27 11:20:27 -07:00
Glenn Jocher 4a4bfb20de FLOPS verbose=False 2020-04-26 16:31:57 -07:00
Glenn Jocher a0a3bab9e6 add Mish() support 2020-04-26 16:31:21 -07:00
Glenn Jocher 18d4ebfd12 add Mish() support 2020-04-26 16:25:46 -07:00
Glenn Jocher 11f228eb00 yolov4.cfg from alexeyab/darknet 2020-04-26 16:07:29 -07:00
Glenn Jocher 55757421de remove future 2020-04-26 14:09:12 -07:00
Glenn Jocher efbeb283c4 ONNX grid float 2020-04-26 14:01:20 -07:00
Glenn Jocher 3bf0cb9c60 remove tb-nightly 2020-04-26 12:41:15 -07:00
Glenn Jocher daedfc5487 reduce merge limit to 3000 2020-04-26 11:22:29 -07:00
Glenn Jocher 754a1b5bf8 reduce merge limit to 3000 2020-04-25 21:15:30 -07:00
Glenn Jocher 3554ab07fb anchor correction 2020-04-24 11:00:01 -07:00
Glenn Jocher 5a8efa5c1d auto --accumulate 2020-04-23 14:32:28 -07:00
Glenn Jocher b3dfd89878 scheduler resume bug fix 2020-04-23 10:35:08 -07:00
Glenn Jocher c29be7f85d torch >= 1.5 2020-04-22 17:55:23 -07:00
Glenn Jocher aa854ecaa9 torch >= 1.5 2020-04-22 17:54:56 -07:00
Glenn Jocher dda0afa22e onnx export IO layer names update 2020-04-22 16:00:20 -07:00
Glenn Jocher 748f60baae updated test default img_size 512 2020-04-22 14:32:51 -07:00
Glenn Jocher 345d65d18f updated train default img_size 320-640 2020-04-22 14:32:05 -07:00
Glenn Jocher 82a12e2c8e docker train update 2020-04-22 11:38:48 -07:00
Glenn Jocher 7cbac5a3ea train.py iou_t to 0.20 2020-04-22 11:34:34 -07:00
Glenn Jocher 77b3829d56 check_git_status() to train.py 2020-04-22 11:02:09 -07:00
Glenn Jocher 2f636d5740 .half() bug fix 2020-04-22 10:46:26 -07:00
Glenn Jocher 03c6a2d6fa cleanup 2020-04-21 12:31:37 -07:00
Glenn Jocher a5bd0fe567 tensorboard comment=opt.name 2020-04-21 12:19:29 -07:00
Glenn Jocher 4c4f4f4dd4 Merge remote-tracking branch 'origin/master' 2020-04-21 09:14:15 -07:00
Glenn Jocher 6aed6c5fd0 attempt_download() update for '' weights 2020-04-21 09:13:38 -07:00
Glenn Jocher 22a6c441ce Update README.md 2020-04-21 00:39:06 -07:00
Glenn Jocher 8b45360e28 detect cleanup 2020-04-20 16:47:28 -07:00
Glenn Jocher cdb69d5929 cfg cleanup 2020-04-20 16:34:00 -07:00
Glenn Jocher be3f322375 Tensorboard out of try, iou_t to 0.10 2020-04-20 09:57:15 -07:00
Glenn Jocher accce6b565 git status check bug fix 2020-04-18 18:06:11 -07:00
Glenn Jocher 693c06b26c bug fix issues/1067 2020-04-18 12:07:44 -07:00
Glenn Jocher bf1061c146 cleanup 2020-04-16 16:12:23 -07:00
Glenn Jocher 9ea856242f Merge remote-tracking branch 'origin/master' 2020-04-15 22:03:55 -07:00
Glenn Jocher c3edf8daf4 move image size report 2020-04-15 22:03:51 -07:00
Glenn Jocher 716e618a18 Update greetings.yml 2020-04-15 13:14:54 -07:00
Glenn Jocher 6566f37cdc Merge remote-tracking branch 'origin/master' 2020-04-15 12:55:00 -07:00
Glenn Jocher 510eadcfa5 Apex and 'git pull' suggestions 2020-04-15 12:54:56 -07:00
Glenn Jocher 20a094ccb9 Update README.md 2020-04-15 12:25:42 -07:00
Glenn Jocher b8c3644a18 ONNX export update 2020-04-15 12:12:59 -07:00
Glenn Jocher 628028c617 bias init 2020-04-15 11:50:54 -07:00
Glenn Jocher a49ea80218 update initialize_weights() 2020-04-14 15:58:32 -07:00
Glenn Jocher ac4c90c817 cleanup 2020-04-14 13:08:00 -07:00
Glenn Jocher f5a2682a81 image sizes report 2020-04-14 12:02:08 -07:00
Glenn Jocher 763cdd5ae2 detailed image sizes report 2020-04-14 11:51:19 -07:00
Glenn Jocher 029e137bc2 bug fix 2020-04-14 04:34:40 -07:00
Glenn Jocher 1681249588 cleanup 2020-04-14 04:15:53 -07:00
Glenn Jocher 198a5a591d code cleanup 2020-04-14 04:15:05 -07:00
Glenn Jocher 25725c8569 bug fix 2020-04-14 03:13:30 -07:00
Glenn Jocher 835b0da68a new modules and init weights 2020-04-14 01:20:57 -07:00
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