Skip to content

Commit 79201d6

Browse files
authored
code cleanup (#64)
* code cleanup resolves #52
1 parent 930cc63 commit 79201d6

File tree

103 files changed

+221
-21418
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

103 files changed

+221
-21418
lines changed

.travis.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ install:
88
- docker run -it -d --rm -p 5000:5000 max-object-detector
99
- pip install -r requirements-test.txt
1010
before_script:
11-
- flake8 . --max-line-length=127 --exclude utils/,protos/
11+
- flake8 . --max-line-length=127
1212
- sleep 30
1313
script:
1414
- pytest tests/test.py

core/model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import logging
2424
from config import PATH_TO_CKPT, PATH_TO_LABELS, NUM_CLASSES
2525
from utils import label_map_util
26+
import utils.ops
2627

2728
logger = logging.getLogger()
2829

@@ -89,7 +90,7 @@ def _predict(self, imageRaw, threshold): # was originally run_inference_for_sin
8990
real_num_detection = tf.cast(tensor_dict['num_detections'][0], tf.int32)
9091
detection_boxes = tf.slice(detection_boxes, [0, 0], [real_num_detection, -1])
9192
detection_masks = tf.slice(detection_masks, [0, 0, 0], [real_num_detection, -1, -1])
92-
detection_masks_reframed = utils_ops.reframe_box_masks_to_image_masks(detection_masks, # noqa #TODO
93+
detection_masks_reframed = utils.ops.reframe_box_masks_to_image_masks(detection_masks,
9394
detection_boxes,
9495
image.shape[0],
9596
image.shape[1])

0 commit comments

Comments
 (0)