Images fed to the network in demo.py are BGR instead of RGB. This is apparent in the image in your readme - the car's headlights are blue, as are the bricks in the building. Please add this cvtColor in demo.py:
def process(self, file_name=None, img=None, show=False):
if file_name is not None:
img = cv2.imread(file_name)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
Images fed to the network in demo.py are BGR instead of RGB. This is apparent in the image in your readme - the car's headlights are blue, as are the bricks in the building. Please add this
cvtColor
indemo.py
: