@@ -10,6 +10,7 @@ import re
|
|||||||
import warnings
|
import warnings
|
||||||
import cssutils
|
import cssutils
|
||||||
import logging
|
import logging
|
||||||
|
import matplotlib.colors
|
||||||
cssutils.log.setLevel(logging.ERROR)
|
cssutils.log.setLevel(logging.ERROR)
|
||||||
|
|
||||||
def remove_namespaces(s):
|
def remove_namespaces(s):
|
||||||
@@ -70,6 +71,10 @@ def parse_color(s, defs):
|
|||||||
elif s == 'none':
|
elif s == 'none':
|
||||||
return None
|
return None
|
||||||
else:
|
else:
|
||||||
|
try :
|
||||||
|
rgba = matplotlib.colors.to_rgba(s)
|
||||||
|
color = torch.tensor(rgba)
|
||||||
|
except ValueError :
|
||||||
warnings.warn('Unknown color command ' + s)
|
warnings.warn('Unknown color command ' + s)
|
||||||
return color
|
return color
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user