def save_segmentation_folder(xyzc, file, mask_class = 16,res = (256, 256), out_path=""): out_folder = out_path #shutil.rmtree(out_folder, ignore_errors=True) #os.mkdir(out_folder) img_tiles, mask_tiles, tile_coord = get_zenith_tiles(xyzc, pixel_size=0.2, res=res, mask_class=mask_class) path = "%s/%s" % (out_folder, file) JSONUtils.makeDirectory(path + "/images/") JSONUtils.makeDirectory(path + "/masks_class_%d/" % mask_class) min_pixels = (res[0] * res[1]) * 0.01 for i in range(len(img_tiles)): img = img_tiles[i] if np.sum(img != 0) > min_pixels: image_name = "%s_%d.png" % (file, i) mask_name = "%s_mask_%d.png" % (file, i) filename = "%s/images/%s" % (path, image_name) imageio.imwrite(filename, img) filename = "%s/masks/%s" % (path, mask_name) imageio.imwrite(filename, mask_tiles[i]) tile_coord[i]["imageName"] = image_name tile_coord[i]["number"] = i tile_coord = [t for t in tile_coord if "imageName" in t] JSONUtils.writeJSON(tile_coord, "%s/images_index.json" % path) return tile_coord
def retrieveCardDataIntoVector(cardNumber, cardDicts,useRelevant): vector = [] cardName = JSONUtils.getCardNameFromCardNumber(cardNumber,useRelevant) if cardName == None: return None for dict in cardDicts: if cardName in dict: vector.append(int(dict[cardName])) else: vector.append(0) return vector
def testVectorRetrieval(lookupDicts): print 'Starting Vector Retrieval for all cards...' allCardsVectors = [] for x in range(1,30000): if x%500 == 0: print 'Processed '+str(x)+' cards...' v = retrieveCardDataIntoVector(x, lookupDicts, False) if v != None: v = [x] + v allCardsVectors.append(v) print 'Done! Printing cards that were in a tournament deck... \n' for cardVector in allCardsVectors: if sum(cardVector[1:]) > 0: print '{:<60}'.format(JSONUtils.getCardNameFromCardNumber(cardVector[0],False))\ + str(cardVector[1:]) print 'All done!'
def verifyMappingJSONToDicts(lookupDicts): JSONCardsDict = JSONUtils.parseJsonIntoCardsDictionary() errorCount = {} for (i,dict) in enumerate(lookupDicts): tourneyInfo = 'Tournament:'+str((i-(i%5))/5+1) if (i%5 ==0): print 'Processing ' + tourneyInfo + '...' for (idx,key) in enumerate(dict): if key not in JSONCardsDict.keys(): thiskey = tourneyInfo+': '+key if thiskey not in errorCount: errorCount[thiskey] = 1 else: errorCount[thiskey] += 1 print '\n' for key in errorCount: print key + ':' + str(errorCount[key])
def handle(params): try: mode = urllib.unquote_plus(params["mode"]) except: mode = None dialog = xbmcgui.Dialog() if mode == 'mute-movie': movieid = urllib.unquote_plus(params["id"]) xbmc.log("movieid: %s" % str(movieid)) details = data.getMovieDetails(movieid) xbmc.log('details: %s' % details) ret = dialog.yesno(details['label'], Addon.getLocalizedString(30302)) if not ret: sys.exit() fileLoc = details['file'] finder = subf.SubFinder(Addon) srtLoc = finder.getSRT(fileLoc) if srtLoc: xbmc.log("Using srt file: %s" % srtLoc) if createEDL(srtLoc, fileLoc): dialog.ok(details['label'], Addon.getLocalizedString(30303)) else: dialog.ok(details['label'], Addon.getLocalizedString(30307)) elif mode == 'mute-episode': episodeId = urllib.unquote_plus(params["id"]) xbmc.log("episodeId: %s" % str(episodeId)) details = data.getEpisodeDetails(episodeId) xbmc.log('details: %s' % details) ret = dialog.yesno(details['showtitle'], Addon.getLocalizedString(30308)) if not ret: sys.exit() fileLoc = details['file'] finder = subf.SubFinder(Addon) srtLoc = finder.getSRT(fileLoc) if srtLoc: xbmc.log("Using srt file: %s" % srtLoc) if createEDL(srtLoc): dialog.ok(details['showtitle'], Addon.getLocalizedString(30309)) else: dialog.ok(details['showtitle'], Addon.getLocalizedString(30307)) elif mode == 'movies': movieDict = data.GetAllMovies() xbmc.log("movieDict: %s" % movieDict) listing = [] for movie in movieDict: url = sys.argv[0] + "?mode=mute-movie&id=" + str(movie['movieid']) listing.append([movie['label'], movie['thumbnail'], url]) showThumbnailListing(listing) elif mode == 'tv': shows = data.GetAllTVShows() xbmc.log('tv shows: %s' % shows) listing = [] for show in shows: url = sys.argv[0] + "?mode=tvshow-details&id=" + str(show['tvshowid']) listing.append([show['label'], show['thumbnail'], url]) showThumbnailListing(listing, True) elif mode == 'tvshow-details': showId = urllib.unquote_plus(params["id"]) episodes = data.GetTVShowEpisodes(showId) xbmc.log('episodes: %s' % episodes) listing = [] for episode in episodes: url = sys.argv[0] + "?mode=mute-episode&id=" + str(episode['episodeid']) listing.append([episode['label'], episode['thumbnail'], url]) showThumbnailListing(listing, True) elif mode == 'view-active': show = dialog.yesno("WARNING", "This will show offensive words on screen.\nAre you sure you want to view it now?") if show: showSimpleListing(getWordsListing()) elif mode == 'show-all': show = dialog.yesno("WARNING", "This will show offensive words on screen.\nAre you sure you want to view it now?") if show: dialog.ok('XBMC', "To edit this list, modify 'filter.txt' in the script's directory" ) showSimpleListing(getWordsListing(True)) elif mode == 'word-details': categoryId = int(urllib.unquote_plus(params["categoryId"])) wordId = int(urllib.unquote_plus(params["wordId"])) categories = get_categories() prefs = get_severities() category = categories.keys()[categoryId] word, severity = categories[category][wordId] isFiltered = filter.is_blocked(severity, category, prefs) blockNum = None if category in prefs: blockNum = prefs[category] title = 'Blocking This Word' if isFiltered else 'Not Actively Blocking' line0 = 'Category: %s' % category line1 = "Severity (1 is lowest, 10 is most offensive): %d" % severity if blockNum: line2 = 'Current category setting blocks words with severity >= %d' % blockNum else: line2 = 'No setting for this category, cannot block it' dialog.ok(title, line0, line1, line2) else: xbmc.log("Running %s v%s. Using default listing." % (__scriptid__, __version__)) xbmc.log(" %s " % str(sys.argv[:])) showSimpleListing(getInitialListing())
def handle(params): try: mode = urllib.unquote_plus(params["mode"]) except: mode = None dialog = xbmcgui.Dialog() if mode == 'mute-movie': movieid = urllib.unquote_plus(params["id"]) xbmc.log("movieid: %s" % str(movieid)) details = data.getMovieDetails(movieid) xbmc.log('details: %s' % details) ret = dialog.yesno(details['label'], Addon.getLocalizedString(30302)) if not ret: sys.exit() fileLoc = details['file'] finder = subf.SubFinder(Addon) srtLoc = finder.getSRT(fileLoc) if srtLoc: xbmc.log("Using srt file: %s" % srtLoc) if createEDL(srtLoc, fileLoc): dialog.ok(details['label'], Addon.getLocalizedString(30303)) else: dialog.ok(details['label'], Addon.getLocalizedString(30307)) elif mode == 'mute-episode': episodeId = urllib.unquote_plus(params["id"]) xbmc.log("episodeId: %s" % str(episodeId)) details = data.getEpisodeDetails(episodeId) xbmc.log('details: %s' % details) ret = dialog.yesno(details['showtitle'], Addon.getLocalizedString(30308)) if not ret: sys.exit() fileLoc = details['file'] finder = subf.SubFinder(Addon) srtLoc = finder.getSRT(fileLoc) if srtLoc: xbmc.log("Using srt file: %s" % srtLoc) if createEDL(srtLoc, fileLoc): dialog.ok(details['showtitle'], Addon.getLocalizedString(30309)) else: dialog.ok(details['showtitle'], Addon.getLocalizedString(30307)) elif mode == 'movies': movieDict = data.GetAllMovies() xbmc.log("movieDict: %s" % movieDict) listing = [] for movie in movieDict: url = sys.argv[0] + "?mode=mute-movie&id=" + str(movie['movieid']) listing.append([movie['label'], movie['thumbnail'], url]) showThumbnailListing(listing) elif mode == 'tv': shows = data.GetAllTVShows() xbmc.log('tv shows: %s' % shows) listing = [] for show in shows: url = sys.argv[0] + "?mode=tvshow-details&id=" + str( show['tvshowid']) listing.append([show['label'], show['thumbnail'], url]) showThumbnailListing(listing, True) elif mode == 'tvshow-details': showId = urllib.unquote_plus(params["id"]) episodes = data.GetTVShowEpisodes(showId) xbmc.log('episodes: %s' % episodes) listing = [] for episode in episodes: url = sys.argv[0] + "?mode=mute-episode&id=" + str( episode['episodeid']) listing.append([episode['label'], episode['thumbnail'], url]) showThumbnailListing(listing, True) elif mode == 'view-active': show = dialog.yesno( "WARNING", "This will show offensive words on screen.\nAre you sure you want to view it now?" ) if show: showSimpleListing(getWordsListing()) elif mode == 'show-all': show = dialog.yesno( "WARNING", "This will show offensive words on screen.\nAre you sure you want to view it now?" ) if show: dialog.ok( 'XBMC', "To edit this list, modify 'filter.txt' in the script's directory" ) showSimpleListing(getWordsListing(True)) elif mode == 'word-details': categoryId = int(urllib.unquote_plus(params["categoryId"])) wordId = int(urllib.unquote_plus(params["wordId"])) categories = get_categories() prefs = get_severities() category = categories.keys()[categoryId] word, severity = categories[category][wordId] isFiltered = filter.is_blocked(severity, category, prefs) blockNum = None if category in prefs: blockNum = prefs[category] title = 'Blocking This Word' if isFiltered else 'Not Actively Blocking' line0 = 'Category: %s' % category line1 = "Severity (1 is lowest, 10 is most offensive): %d" % severity if blockNum: line2 = 'Current category setting blocks words with severity >= %d' % blockNum else: line2 = 'No setting for this category, cannot block it' dialog.ok(title, line0, line1, line2) else: xbmc.log("Running %s v%s. Using default listing." % (__scriptid__, __version__)) xbmc.log(" %s " % str(sys.argv[:])) showSimpleListing(getInitialListing())
import skimage.io as io file = "000029.las" path = "Images/%s" % file xyzic = tools.read_las(file) xyzc = xyzic[:, [0,1,2,4]] img_tiles, mask_tiles, tile_coord = tools.get_zenith_feature_tiles(xyzic, pixel_size=0.2, res=(256, 256), mask_class=16) if False: index = tools.save_tiles_folder(xyzc, file, out_path="Images/") else: index = JSONUtils.readJSON(filename="%s/images_index.json" % path) if False: final_img = tools.predict_tiles(index, path=path) imageio.imwrite("%s/predicted.png" % path, final_img) else: final_img = io.imread("%s/predicted.png" % path, as_gray=True) if True: ps = tools.calculate_probability(xyzc[:,0:3], final_img, pixel_size=0.2) print(ps.shape) sio.savemat("results.mat", {"predictions": ps}) results = sio.loadmat("results.mat") predictions = (255.0 - results["predictions"]) / 255.0
if srt.createNewSRT(): os.rename(srtLoc, srtLoc + '.bck') os.rename(srtLoc[:-3] + "tmp", srtLoc) return True except: return False try: mode=urllib.unquote_plus(params["mode"]) except: mode = None if mode == 'movie-details': movieid=urllib.unquote_plus(params["id"]) xbmc.log("movieid: %s" % str(movieid)) details = data.getMovieDetails(movieid) xbmc.log('details: %s' % details) dialog = xbmcgui.Dialog() ret = dialog.yesno(details['label'], plugin.get_string(30302)) if not ret: sys.exit() fileLoc = details['file'] finder = subf.SubFinder(Addon, plugin) srtLoc = finder.getSRT(fileLoc) if srtLoc: xbmc.log("Using srt file: %s" % srtLoc) if createEDL(): dialog.ok(details['label'], plugin.get_string(30303)) else: