Fit code

Author: cindygr

Image_based/HandleFileNames.py

@@ -35,6 +35,7 @@
 import json
 from os.path import exists, isdir
 from os import mkdir
+import numpy as np
 
 
 class HandleFileNames:
@@ -375,6 +376,97 @@ def read_filenames(fname, path=None):
 
         return handle_files
 
+    @staticmethod
+    def get_name_base_no_number(fname):
+        """Take out everything after ., and take out numbers
+        @param fname - the file name
+        @return the file name, stripped"""
+        name_pieces1 = fname.split(".")[0]
+        name_pieces2 = name_pieces1.split("_")[0]
+        name = "rbg"
+        for s in name_pieces2:
+            if 
+        name_pieces_prev = fnames_images[im_prev]
+
+def make_blueberry_dataset(path_src, path_dest, img_type="jpg", n_each_folder=2, pair_spacing=0):
+    """Assumes there's a folder with multiple folders with multiple images in each folder
+    Grab n_each_folder evenly spaced (time wise) for each director, either single or pairs
+    @param path_src - directory that has the directories
+    @param path_dest - directory to put the images in (will put images from each in a sub folder)
+    @param img_type - one of jpg, png, etc
+    @param n_each_folder - how many frames to grab from each sub folder
+    @param pair_spacing - grab pairs of images, spaced n apart in time (if zero, only grabs one image)"""
+
+    from os import system
+
+    if not exists(path_src):
+        print(f"No directory {path_src} exists, bailing")
+        return
+
+    if not exists(path_dest):
+        mkdir(path_dest)
+
+    search_path = f"{path_src}/*"
+    fnames = glob(search_path)
+    if fnames is None:
+        raise ValueError(f"No sub directories in directory {search_path}")
+
+    # Look in path_src for the sub directories
+    fnames.sort()
+    # Handles the case where the files are in the given directory
+    fnames.append(".")
+    for n in fnames:
+        if not isdir(n):
+            continue
+
+        # List of images
+        search_dir_path = f"{path_src}/{n}/*.{img_type}"
+        fnames_images = glob(search_dir_path)
+        path_dest_subdir = f"{path_dest}/"
+        b_use_sub_folder_names = False
+        if fnames_images is None:
+            search_dir_path = f"{path_src}/{n}/color/*.{img_type}"
+            fnames_images = glob(search_dir_path)
+            path_dest_subdir = f"{path_dest}/{n}/"
+            b_use_sub_folder_names = True
+
+        if fnames_images is None:
+            print("Subdir {n} has no images of type {img_type}")
+            continue;
+        
+        if not exists(path_dest_subdir):
+            if b_use_sub_folder_names:
+                mkdir(path_dest_subdir)
+
+        # Copy the images over first
+        fnames_images.sort()
+        im_step = len(fnames_images) // n_each_folder
+        im_keep = np.linspace(im_step // 2, len(fnames_images), n_each_folder)
+        for im in im_keep:
+            im_i = int(im)
+            im_prev = max(0, im_i - pair_spacing // 2)
+            im_next = min(im_prev + pair_spacing, len(fnames_images) - 1)
+
+            name_pieces_prev = fnames_images[im_prev].split("_")
+            name_prev = ""
+            for s in name_pieces_prev:
+            name_pieces_prev = fnames_images[im_prev]
+            sys_cmd_str = f"cp {search_dir_path}/{fnames_images[im_prev]} {path_dest_subdir}{fnames_images[im_prev]}"
+            system(sys_cmd_str)
+
+            if im_next != im_prev:
+                sys_cmd_str = f"cp {search_dir_path}/{fnames_images[im_next]} {path_dest_subdir}{fnames_images[im_next]}"
+                system(sys_cmd_str)
+
+        if b_use_sub_folder_names:
+            all_files.add
+
+
+
+    all_files = HandleFileNames(path_dest, img_type=img_type)
+
+
+
 
 if __name__ == '__main__':
     # Example bb

Image_based/fit_bezier_cyl_2d_mask.py

@@ -204,6 +204,8 @@ def score_mask_fit(self, im_mask):
 
 
 if __name__ == '__main__':
+    im = cv2.imread("/Users/grimmc/Downloads/depth_raw_3.jpg")
+    print(f" {im.min()}, {im.max()}")
     # path_bpd = "./data/trunk_segmentation_names.json"
     path_bpd = "./data/forcindy_fnames.json"
     all_files = HandleFileNames.read_filenames(path_bpd)