Visitar URL original
Exclude labeled folders and change of codec · mathang/DeepLabCut@08fc7b2 · GitHub
Skip to content

Commit 08fc7b2

Browse files
committed
Exclude labeled folders and change of codec
1 parent 9019655 commit 08fc7b2

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

‎deeplabcut/refine_training_dataset/outlier_frames.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,7 @@ def merge_datasets(config,forceiterate=None):
487487
config_path = Path(config).parents[0]
488488

489489
bf=Path(str(config_path/'labeled-data'))
490-
allfolders = [os.path.join(bf,fn) for fn in os.listdir(bf) ]
490+
allfolders = [os.path.join(bf,fn) for fn in os.listdir(bf) and "_labeled" not in fn] #exclude labeled data folders!
491491
flagged=False
492492
for findex,folder in enumerate(allfolders):
493493
if os.path.isfile(os.path.join(folder,'MachineLabelsRefine.h5')): #Folder that was manually refine...

‎deeplabcut/utils/make_labeled_video.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def CreateVideoSlow(clip,Dataframe,tmpfolder,dotsize,colormap,alphavalue,pcutoff
147147
os.remove(file_name)
148148
os.chdir(start)
149149

150-
def create_labeled_video(config,videos,shuffle=1,trainingsetindex=0,videotype='avi',save_frames=False,delete=False,displayedbodyparts='all',codec='X264'):
150+
def create_labeled_video(config,videos,shuffle=1,trainingsetindex=0,videotype='avi',save_frames=False,delete=False,displayedbodyparts='all',codec='avc1'):
151151
"""
152152
Labels the bodyparts in a video. Make sure the video is already analyzed by the function 'analyze_video'
153153

0 commit comments

Comments
 (0)