Skip to content

Commit 790b635

Browse files
committed
minor modifications
1 parent d11c36a commit 790b635

File tree

2 files changed

+4
-10
lines changed

2 files changed

+4
-10
lines changed

README.md

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55
This is my implementation of [YOLOv3](https://pjreddie.com/media/files/papers/YOLOv3.pdf) in pure TensorFlow. It contains the full pipeline of training and evaluation on your own dataset. The keys features of this repo are:
66

77
- Efficient tf.data pipeline
8-
98
- Weights converter (converting pretrained darknet weights on COCO dataset to TensorFlow checkpoint.)
109
- Extremely fast GPU non maximum supression.
1110
- Full training pipeline.
1211
- Kmeans algorithm to select prior anchor boxes.
12+
- [ ] Multi-GPU training with sync batch norm. (on working)
1313

1414
### 2. Requirements
1515

@@ -117,8 +117,7 @@ Using `train.py`. The parameters are as following:
117117

118118
```shell
119119
$ python train.py -h
120-
usage: train.py [-h] [--train_file TRAIN_FILE]
121-
[--val_file VAL_FILE]
120+
usage: train.py [-h] [--train_file TRAIN_FILE] [--val_file VAL_FILE]
122121
[--restore_path RESTORE_PATH]
123122
[--save_dir SAVE_DIR]
124123
[--log_dir LOG_DIR]
@@ -161,8 +160,7 @@ Using `eval.py` to evaluate the validation or test dataset. The parameters are a
161160
162161
```shell
163162
$ python eval.py -h
164-
usage: eval.py [-h] [--eval_file EVAL_FILE]
165-
[--restore_path RESTORE_PATH]
163+
usage: eval.py [-h] [--eval_file EVAL_FILE] [--restore_path RESTORE_PATH]
166164
[--anchor_path ANCHOR_PATH]
167165
[--class_name_path CLASS_NAME_PATH]
168166
[--batch_size BATCH_SIZE]
@@ -188,10 +186,6 @@ There are many skills you can try during training:
188186
189187
(2) Mutil-scale training: You can change the input image scales periodically like the author does in the original paper.
190188
191-
### 9. TODO
192-
193-
- [ ] Multi-GPU training with sync batch norm.
194-
195189
-------
196190
197191
### Credits:

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,6 @@
261261
writer.add_summary(make_summary('evaluation/val_recall', recall), global_step=epoch)
262262
writer.add_summary(make_summary('evaluation/val_precision', precision), global_step=epoch)
263263

264-
# manually shuffle the training data in a new wpoch
264+
# manually shuffle the training data in a new epoch
265265
shuffle_and_overwrite(args.train_file)
266266
sess.run(train_iterator.initializer)

0 commit comments

Comments
 (0)