Commit Graph

  • 6736d7d125 swap cv2.INTER_AREA for cv2.INTER_LINEAR Glenn Jocher 2020-04-10 12:47:07 -07:00
  • b98ce11d3a add MixConv2d() layer Glenn Jocher 2020-04-09 20:20:23 -07:00
  • 6e19245dc8 auto strip optimizer from best.pt after training Glenn Jocher 2020-04-09 19:53:29 -07:00
  • bc74822540 notebook update Glenn Jocher 2020-04-09 14:33:24 -07:00
  • d1601ae0f3 training updates Glenn Jocher 2020-04-08 21:34:34 -07:00
  • 4120ac3aa6 training updates Glenn Jocher 2020-04-08 21:01:58 -07:00
  • c7f5b6cc21 speed update Glenn Jocher 2020-04-08 20:53:17 -07:00
  • d9af081b7f Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-04-08 20:43:57 -07:00
  • 8f71b7865b run once to remove initial timing effects Glenn Jocher 2020-04-08 20:43:51 -07:00
  • 4f6843e249
    Update README.md Glenn Jocher 2020-04-08 12:28:10 -07:00
  • 13fa2798a6
    Update README.md Glenn Jocher 2020-04-08 12:27:33 -07:00
  • b6959a2f54 ONNX export self.training=False Glenn Jocher 2020-04-08 10:25:52 -07:00
  • 97780cfdb4 parameterize grid size Glenn Jocher 2020-04-08 10:14:33 -07:00
  • 9f41b7601a inference speed and mAP updates Glenn Jocher 2020-04-08 10:02:20 -07:00
  • f54d28ba63 improve assert no labels found Glenn Jocher 2020-04-08 09:57:59 -07:00
  • 933e05cb44 augment update Glenn Jocher 2020-04-07 17:35:35 -07:00
  • 6c5ecaf805 remove label loading during training Glenn Jocher 2020-04-07 16:57:22 -07:00
  • b3e1d74478 remove imwrite from augment Glenn Jocher 2020-04-07 14:23:31 -07:00
  • d79c3bd076 parameterize augment scales Glenn Jocher 2020-04-07 14:19:43 -07:00
  • b9b14bef59 scale_img() bug fix Glenn Jocher 2020-04-07 13:35:47 -07:00
  • 067ee264c0 scale_img() bug fix Glenn Jocher 2020-04-07 12:59:52 -07:00
  • 68f58f4dec scale_img() update Glenn Jocher 2020-04-07 12:51:52 -07:00
  • 1a3c77df95 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-04-07 12:27:56 -07:00
  • 1a511d2906 updated --augment sizes and results Glenn Jocher 2020-04-07 12:27:49 -07:00
  • b20e3c4c40
    Update README, add link to yolov3-spp-tensorrt (#1017) Wang Xinyu 2020-04-07 13:56:39 +08:00
  • 4fa3fd2df3 tensorboard notice Glenn Jocher 2020-04-06 16:33:23 -07:00
  • 05ae6e8499 tensorboard/focal loss reporting update Glenn Jocher 2020-04-06 15:45:18 -07:00
  • c7f93bae40 update coco-tuned hyp['cls'] to current dataset Glenn Jocher 2020-04-06 10:58:07 -07:00
  • 4fc0012829 initial batchnorm to 0.03 momentum Glenn Jocher 2020-04-05 18:03:49 -07:00
  • 26fc4fb018 dataloader default color to imagenet mean 114 Glenn Jocher 2020-04-05 18:02:41 -07:00
  • 2baf4e3f93 imagenet normalization on layer 0 batchnorm2d() Glenn Jocher 2020-04-05 17:33:06 -07:00
  • b70cfa9a29 mAP updates for rect inference 64 commit Glenn Jocher 2020-04-05 17:18:24 -07:00
  • c6d4e80335 move inference augmentation to model.forward() Glenn Jocher 2020-04-05 17:14:26 -07:00
  • 4da5c6c114 rect padding to 64, mAP increase 42.7 to 42.9 Glenn Jocher 2020-04-05 16:06:27 -07:00
  • bb59ffe68f model forward() zip() removal Glenn Jocher 2020-04-05 15:22:32 -07:00
  • a657345b45 add FeatureConcat() module Glenn Jocher 2020-04-05 14:47:41 -07:00
  • 968b2ec004 .fuse() after .eval() Glenn Jocher 2020-04-05 14:05:12 -07:00
  • d04738a27c forward updated if-else Glenn Jocher 2020-04-05 13:49:13 -07:00
  • e81a152a92 tensorboard notice and model verbose option Glenn Jocher 2020-04-05 13:35:58 -07:00
  • a19b1a3b94 line thickness Glenn Jocher 2020-04-05 11:10:05 -07:00
  • 6203340888 detect.py multi_label default False Glenn Jocher 2020-04-05 11:05:49 -07:00
  • 41246aa042 tensorboard updates Glenn Jocher 2020-04-04 19:34:39 -07:00
  • 00c1fdd805 add MixConv2d() layer Glenn Jocher 2020-04-03 20:03:44 -07:00
  • eb9fb245aa add support for standalone BatchNorm2d() Glenn Jocher 2020-04-03 14:21:47 -07:00
  • 682c2b27e7 smart bias bug fix Glenn Jocher 2020-04-03 12:42:09 -07:00
  • 41a002e798 grid.float() Glenn Jocher 2020-04-03 12:38:08 -07:00
  • 93055a9d58 create_grids() to YOLOLayer method Glenn Jocher 2020-04-02 20:23:55 -07:00
  • 91f563c2a2 create_grids() to YOLOLayer method Glenn Jocher 2020-04-02 19:10:51 -07:00
  • 207c6fcff9 merge NMS full matrix Glenn Jocher 2020-04-02 18:53:40 -07:00
  • aa4591d7e9 batchnorm momentum to 0.03 Glenn Jocher 2020-04-02 14:10:45 -07:00
  • 9155ef3f4f burnin merged with prebias Glenn Jocher 2020-04-02 14:08:21 -07:00
  • 27c7334e81 new layers.py file Glenn Jocher 2020-04-02 12:22:15 -07:00
  • 4ac60018f6 FLOPS report Glenn Jocher 2020-04-01 14:05:41 -07:00
  • ea80ba65af documentation updates Glenn Jocher 2020-04-01 12:16:37 -07:00
  • 765b8f3a3b documentation update Glenn Jocher 2020-04-01 12:12:14 -07:00
  • 5b322b6038 nvcr update Glenn Jocher 2020-04-01 09:57:00 -07:00
  • 300e9a7ad6 merge NMS full matrix Glenn Jocher 2020-03-31 21:31:09 -07:00
  • 8d788e10c4 mAP updates Glenn Jocher 2020-03-31 19:07:41 -07:00
  • 02802e67f2 merge NMS full matrix Glenn Jocher 2020-03-31 18:18:08 -07:00
  • f4eecef700 merge NMS speed/memory improvements Glenn Jocher 2020-03-31 15:37:23 -07:00
  • 992e0d7cb4 default test --conf to 0.001 Glenn Jocher 2020-03-31 14:36:25 -07:00
  • 98271eb6ed remove deprecated models Glenn Jocher 2020-03-31 14:27:10 -07:00
  • 16862ea846 update 'reproduce our results' Glenn Jocher 2020-03-30 21:21:45 -07:00
  • b2d9f1898f burnin lr ramp 300 iterations Glenn Jocher 2020-03-30 19:27:42 -07:00
  • ac2aa56e0a feature fusion update Glenn Jocher 2020-03-30 17:53:17 -07:00
  • 108334db29 FLOPs update Glenn Jocher 2020-03-30 16:04:08 -07:00
  • 105882b3c6 GFLOPs correction Glenn Jocher 2020-03-30 15:30:53 -07:00
  • de52a008a5 default --img-size to 512 Glenn Jocher 2020-03-30 11:46:20 -07:00
  • f6fc9634ab mAP updates Glenn Jocher 2020-03-30 11:37:38 -07:00
  • eb151a881e NMS and test batch_size updates Glenn Jocher 2020-03-29 20:41:32 -07:00
  • c6b59a0e8a LR schedule to 0.05 min Glenn Jocher 2020-03-29 13:29:06 -07:00
  • 9c5e76b93d EMA implemented by default Glenn Jocher 2020-03-29 13:14:54 -07:00
  • dc8e56b9f3 mAP update Glenn Jocher 2020-03-28 16:03:46 -07:00
  • ce17c26759 mAP updates Glenn Jocher 2020-03-27 13:52:07 -07:00
  • f9d34587da Merge NMS update Glenn Jocher 2020-03-27 13:11:24 -07:00
  • 582de735ad
    utils.clip_coords doesn't work as expected. (#961) GoogleWiki 2020-03-28 04:09:10 +08:00
  • 4a63b24b09 Merge NMS update Glenn Jocher 2020-03-26 19:50:29 -07:00
  • 01ee0c5e95 Merge NMS update Glenn Jocher 2020-03-26 18:41:04 -07:00
  • dad59220f1 speed and comments update Glenn Jocher 2020-03-26 18:34:20 -07:00
  • faab52913c mAP updates Glenn Jocher 2020-03-26 16:35:46 -07:00
  • 9568d4562d mAP updates Glenn Jocher 2020-03-26 16:22:58 -07:00
  • 5ab13e5aa2 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-03-26 16:20:11 -07:00
  • 470371ba59 Test augment update Glenn Jocher 2020-03-26 16:20:06 -07:00
  • 20b2671de0
    cleanup (#963) Yonghye Kwon 2020-03-27 06:46:17 +09:00
  • a4721e90f8 defult batch-size to 16 Glenn Jocher 2020-03-26 14:22:59 -07:00
  • 4a7d9bdba9 mAP increases Glenn Jocher 2020-03-26 14:14:52 -07:00
  • a322fc5d4b Merge NMS update Glenn Jocher 2020-03-26 12:48:00 -07:00
  • 171b4129b5 Merge NMS update Glenn Jocher 2020-03-26 12:33:12 -07:00
  • eac07f9da3 Merge NMS update Glenn Jocher 2020-03-26 12:20:01 -07:00
  • 94344f5bea test augmentation comments Glenn Jocher 2020-03-26 11:34:32 -07:00
  • f91b1fb13a merge_batch NMS method Glenn Jocher 2020-03-26 11:28:46 -07:00
  • c71ab7d506 augmented testing Glenn Jocher 2020-03-26 11:25:44 -07:00
  • 23b34f4db8 merge_batch NMS method Glenn Jocher 2020-03-25 23:29:33 -07:00
  • aa0c64b5ac merge_batch NMS method Glenn Jocher 2020-03-25 23:24:57 -07:00
  • 3265d50f69 speed update Glenn Jocher 2020-03-19 18:15:09 -07:00
  • 89b6377723 Fuse by default when test.py called directly (faster) Glenn Jocher 2020-03-19 18:11:08 -07:00
  • fff45c39a8 cleanup/speedup Glenn Jocher 2020-03-19 16:41:42 -07:00
  • 1b68fe7fde cleanup Glenn Jocher 2020-03-19 16:23:44 -07:00
  • 83c9cfb7de FocalLoss() and obj loss speed and stability update Glenn Jocher 2020-03-19 12:30:37 -07:00
  • 20454990ce FLOPS report Glenn Jocher 2020-03-19 12:30:07 -07:00