functions update

functions.py

@@ -101,14 +101,7 @@ def crop(image,size=1024):
           del crop_img
           gc.collect()
           #sleep(2)
-    del H
-    del H1
-    del H2
-    del W
-    del W1
-    del W2
-    del h
-    del w
+    del H,H1,H2,W,W1,W2,h,w
     gc.collect()
     sleep(1)
     np.save("locations.npy",np.array(locations))
@@ -129,11 +122,7 @@ def inference(image,locations,model,test_transforms,device):
         with torch.no_grad():
             predictions = model(image_transformed)
 
-        del imgs
-        del name
-        del path
-        del transformed
-        del image_transformed
+        del imgs,name,path,transformed,image_transformed
         gc.collect()
         sleep(1)
 
@@ -153,10 +142,7 @@ def inference(image,locations,model,test_transforms,device):
         label_name = "lab_"+str(location[0])+"_"+str(location[2])+".png"
         cv2.imwrite(os.path.join("labels",label_name),img)
 
-        del label_name
-        del img
-        del nms_prediction
-        del predictions
+        del label_name,img,nms_prediction,predictions
         gc.collect()
         sleep(1)
 
@@ -170,10 +156,7 @@ def create_new_ortho(locations,empty_image):
         empty_image[location[0]:location[1],location[2]:location[3],:] = img
         if i%300==0:
             cv2.imwrite("img.png",empty_image)
-            del img
-            del name
-            del path
-            del empty_image
+            del img,name,path,empty_image
             gc.collect()
             #sleep(1)
             empty_image = np.array(cv2.imread("img.png"))