Commit 2b7a12ef authored by WillBrennan's avatar WillBrennan

Renamed extended to scripts

parent f169e38d
...@@ -7,4 +7,4 @@ __author__ = 'Will Brennan' ...@@ -7,4 +7,4 @@ __author__ = 'Will Brennan'
# Standard Modules # Standard Modules
# Custom Modules # Custom Modules
from main import * from main import *
import extended import scripts
\ No newline at end of file \ No newline at end of file
...@@ -10,7 +10,7 @@ import argparse ...@@ -10,7 +10,7 @@ import argparse
import cv2 import cv2
import numpy import numpy
# Custom Modules # Custom Modules
import extended import scripts
class SkinDetector(object): class SkinDetector(object):
...@@ -140,7 +140,7 @@ class SkinDetector(object): ...@@ -140,7 +140,7 @@ class SkinDetector(object):
def process(image, save=False, display=False, args=None): def process(image, save=False, display=False, args=None):
assert isinstance(image, numpy.ndarray) assert isinstance(image, numpy.ndarray)
if not args: if not args:
args = extended.gen_args() args = scripts.gen_args()
else: else:
assert isinstance(args, argparse.Namespace), 'args must be an argparse.Namespace' assert isinstance(args, argparse.Namespace), 'args must be an argparse.Namespace'
args.save = save args.save = save
...@@ -150,9 +150,9 @@ def process(image, save=False, display=False, args=None): ...@@ -150,9 +150,9 @@ def process(image, save=False, display=False, args=None):
if __name__ == '__main__': if __name__ == '__main__':
args = extended.get_args() args = scripts.get_args()
logger = extended.get_logger(quite=args.quite, debug=args.debug) logger = scripts.get_logger(quite=args.quite, debug=args.debug)
args.image_paths = extended.find_images(args.image_paths[0]) args.image_paths = scripts.find_images(args.image_paths[0])
for image_path in args.image_paths: for image_path in args.image_paths:
img_col = cv2.imread(image_path, 1) img_col = cv2.imread(image_path, 1)
img_msk = process(img_col, args=args) img_msk = process(img_col, args=args)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment