Skip to content

Commit

Permalink
Added yolov3-openimages.cfg and SELU activation
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexeyAB committed Oct 15, 2018
1 parent 2bb4445 commit 76706dc
Show file tree
Hide file tree
Showing 12 changed files with 2,829 additions and 1 deletion.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ This repository supports:
* **GPU with CC >= 3.0**: https://en.wikipedia.org/wiki/CUDA#GPUs_supported

##### Pre-trained models for different cfg-files can be downloaded from (smaller -> faster & lower quality):
* `yolov3-openimages.cfg` (247 MB COCO **Yolo v3**) - requires 4 GB GPU-RAM: https://pjreddie.com/media/files/yolov3-openimages.weights
* `yolov3-spp.cfg` (240 MB COCO **Yolo v3**) - requires 4 GB GPU-RAM: https://pjreddie.com/media/files/yolov3-spp.weights
* `yolov3.cfg` (236 MB COCO **Yolo v3**) - requires 4 GB GPU-RAM: https://pjreddie.com/media/files/yolov3.weights
* `yolov3-tiny.cfg` (34 MB COCO **Yolo v3 tiny**) - requires 1 GB GPU-RAM: https://pjreddie.com/media/files/yolov3-tiny.weights
Expand Down
8 changes: 8 additions & 0 deletions build/darknet/x64/cfg/openimages.data
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
classes= 601
train = /home/pjreddie/data/openimsv4/openimages.train.list
#valid = coco_testdev
valid = data/coco_val_5k.list
names = data/openimages.names
backup = /home/pjreddie/backup/
eval=coco

Loading

0 comments on commit 76706dc

Please sign in to comment.