fix

monoloco/eval/generate_kitti.py

@@ -273,5 +273,4 @@ def create_empty_files(dir_out, net):
 
     for i in range(7481):
         name = "0" * (6 - len(str(i))) + str(i) + '.txt'
-        ff = open(os.path.join(dir_out[net], name), "a+")
-        ff.close()
+        with open(os.path.join(dir_out[net], name), "a+"): pass

monoloco/run.py

@@ -48,7 +48,7 @@ def cli():
     visualizer.cli(parser)
 
     # Monoloco
-    predict_parser.add_argument('--activities', nargs='+', 
+    predict_parser.add_argument('--activities', nargs='+',
                                 help='Choose activities to show: social_distance, raise_hand')
     predict_parser.add_argument('--mode', help='keypoints, mono, stereo', default='mono')
     predict_parser.add_argument('--model', help='path of MonoLoco/MonStereo model to load')

monoloco/utils/kitti.py

@@ -161,8 +161,7 @@ def read_and_rewrite(path_orig, path_new):
                     line_new = temp_1 + ' ' + hwl + ' ' + temp_2 + '\n'
                     ff.write("%s" % line_new)
     except FileNotFoundError:
-        ff = open(path_new, "a+")
-        ff.close()
+        with open(path_new, "a+"): pass
 
 
 def find_cluster(dd, clusters):

monoloco/visuals/pifpaf_show.py

@@ -109,11 +109,11 @@ class KeypointPainter:
             for ci, connection in enumerate(np.array(self.skeleton) - 1):
                 c = color
                 linewidth=self.linewidth
-                if ((connection[0] == 5 and connection[1] == 7) 
+                if ((connection[0] == 5 and connection[1] == 7)
                     or (connection[0] == 7 and connection[1] == 9)) and raise_hand in ['left','both']:
                     c = 'yellow'
                     linewidth = l_arm_width
-                if ((connection[0] == 6 and connection[1] == 8) 
+                if ((connection[0] == 6 and connection[1] == 8)
                     or (connection[0] == 8 and connection[1] == 10)) and raise_hand in ['right', 'both']:
                     c = 'yellow'
                     linewidth = r_arm_width
@@ -192,7 +192,7 @@ class KeypointPainter:
                 matplotlib.patches.Rectangle(
                     (x - scale, y - scale), 2 * scale, 2 * scale, fill=False, color=color))
 
-    def keypoints(self, ax, keypoint_sets, *, 
+    def keypoints(self, ax, keypoint_sets, *,
                   size=None, scores=None, color=None, colors=None, texts=None, raise_hand='none'):
         if keypoint_sets is None:
             return
@@ -443,4 +443,3 @@ def social_distance_colors(colors, dic_out):
     # Prepare color for social distancing
     colors = ['r' if flag else colors[idx] for idx,flag in enumerate(dic_out['social_distance'])]
     return colors
-

monoloco/visuals/printer.py

@@ -248,7 +248,7 @@ class Printer:
             if not self.args.activities or 'social_distance' not in self.args.activities:
                 self.mpl_im0.set_data(image)
 
-        self._front_loop(iterator, axes, number, colors, annotations, dic_out) 
+        self._front_loop(iterator, axes, number, colors, annotations, dic_out)
 
         # Draw the bird figure
         number['num'] = 97