fruit dataset corrected

Author: sonalrpatel

data/fruits/valid/_annotations.txt

@@ -2,6 +2,7 @@ fresh_banana8_jpeg.rf.d0c65fdd8a200236b0888a6b769c3903.jpg 78,131,301,264,46
 fresh_banana35_jpeg.rf.687f032a4b6e7079a27584f2d692192a.jpg 86,77,363,362,46
 fresh_banana48_jpeg.rf.15f50cf86e37055490108357ed72f439.jpg 31,135,149,407,46
 fresh_orange-36-_JPG.rf.693a68ccc89a8ad028beab349685f7a2.jpg 34,82,328,402,49
+fresh_orange-66-_jpg.rf.715315512601a379b8aedeb875381aa5.jpg 119,102,181,186,49 176,87,233,154,49 153,156,216,234,49 244,87,307,173,49 283,215,364,301,49
 fresh_orange-71-_jpg.rf.4654bd63a78b16e47328cf7e81b1f756.jpg 48,68,215,190,49 52,187,214,317,49 195,133,368,268,49
 fresh-apple--12-_jpg.rf.8598b95f486afe00c0033690291fb004.jpg 11,4,380,397,47
 fresh-apple--41-_jpg.rf.c4ce57026cd9deee71c2cf0856bbd13c.jpg 44,41,371,393,47

data/fruits/valid/_classes.txt

@@ -1,20 +1,80 @@
-0
-1
-10
-11
-12
-13
-14
-15
-16
-17
-18
-19
-2
-3
-4
-5
-6
-7
-8
-9
+person
+bicycle
+car
+motorbike
+aeroplane
+bus
+train
+truck
+boat
+traffic light
+fire hydrant
+stop sign
+parking meter
+bench
+bird
+cat
+dog
+horse
+sheep
+cow
+elephant
+bear
+zebra
+giraffe
+backpack
+umbrella
+handbag
+tie
+suitcase
+frisbee
+skis
+snowboard
+sports ball
+kite
+baseball bat
+baseball glove
+skateboard
+surfboard
+tennis racket
+bottle
+wine glass
+cup
+fork
+knife
+spoon
+bowl
+banana
+apple
+sandwich
+orange
+broccoli
+carrot
+hot dog
+pizza
+donut
+cake
+chair
+sofa
+pottedplant
+bed
+diningtable
+toilet
+tvmonitor
+laptop
+mouse
+remote
+keyboard
+cell phone
+microwave
+oven
+toaster
+sink
+refrigerator
+book
+clock
+vase
+scissors
+teddy bear
+hair drier
+toothbrush

train.py

@@ -157,7 +157,7 @@ def _main():
     #   The occupied video memory is large, and all the parameters of the network will be changed
     # =======================================================
     unfreeze_epoch = 50
-    unfreeze_batch_size = int(TRAIN_BATCH_SIZE / 2)
+    unfreeze_batch_size = VAL_BATCH_SIZE
     unfreeze_lr = 1e-4
 
     # =======================================================