Skip to content

Commit f2621b8

Browse files
chore: Change repository structure
1 parent 4d72731 commit f2621b8

File tree

14 files changed

+7
-14
lines changed

14 files changed

+7
-14
lines changed

__init__.py

Whitespace-only changes.

data_generator/__init__.py

Whitespace-only changes.

ssd_batch_generator.py renamed to data_generator/ssd_batch_generator.py

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,6 @@
4444
except ImportError:
4545
warnings.warn("'pickle' module is missing. You won't be able to save parsed file lists and annotations as pickled files.")
4646

47-
# Image processing functions used by the generator to perform the following image manipulations:
48-
# - Translation
49-
# - Horizontal flip
50-
# - Scaling
51-
# - Brightness change
52-
# - Histogram contrast equalization
53-
5447
def _translate(image, horizontal=(0,40), vertical=(0,10)):
5548
'''
5649
Randomly translate the input image horizontally and vertically.
@@ -143,10 +136,10 @@ class BatchGenerator:
143136
144137
Can shuffle the dataset consistently after each complete pass.
145138
146-
Currently provides two methods to parse annotation data: A general-purpose CSV parser
147-
and an XML parser for the Pascal VOC datasets. If the annotations of your dataset are
148-
in a format that is not supported by these parsers, you could just add another parser
149-
method and still use this generator.
139+
Currently provides three methods to parse annotation data: A general-purpose CSV parser,
140+
an XML parser for the Pascal VOC datasets, and a JSON parser for the MS COCO datasets.
141+
If the annotations of your dataset are in a format that is not supported by these parsers,
142+
you could just add another parser method and still use this generator.
150143
151144
Can perform image transformations for data conversion and data augmentation,
152145
for details please refer to the documentation of the `generate()` method.

eval_utils/__init__.py

Whitespace-only changes.

eval_utils/coco_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from math import ceil
2323
import sys
2424

25-
from ssd_box_encode_decode_utils import decode_y
25+
from ssd_box_utils.ssd_box_encode_decode_utils import decode_y
2626

2727
def get_coco_category_maps(annotations_file):
2828
'''

eval_utils/pascal_voc_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from tqdm import trange
2222
import sys
2323

24-
from ssd_box_encode_decode_utils import decode_y
24+
from ssd_box_utils.ssd_box_encode_decode_utils import decode_y
2525

2626
def predict_all_to_txt(model,
2727
img_height,

keras_layers/__init__.py

Whitespace-only changes.

keras_layers/keras_layer_AnchorBoxes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from keras.engine.topology import InputSpec
2424
from keras.engine.topology import Layer
2525

26-
from ssd_box_encode_decode_utils import convert_coordinates
26+
from ssd_box_utils.ssd_box_encode_decode_utils import convert_coordinates
2727

2828
class AnchorBoxes(Layer):
2929
'''

keras_loss_function/__init__.py

Whitespace-only changes.

misc_utils/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)